From e4b3732285d81045da511bbd8e381b7c765ddee9 Mon Sep 17 00:00:00 2001 From: Liv Kelley Date: Sun, 13 Mar 2016 11:39:19 -0400 Subject: [PATCH] This is my code for the pickle counter and the other files I used. --- blah.txt.p | 2 ++ blah2.txt.p | 2 ++ counter.py | 9 ++++++++- 3 files changed, 12 insertions(+), 1 deletion(-) create mode 100644 blah.txt.p create mode 100644 blah2.txt.p diff --git a/blah.txt.p b/blah.txt.p new file mode 100644 index 0000000..ba8c4c4 --- /dev/null +++ b/blah.txt.p @@ -0,0 +1,2 @@ +I3 +. \ No newline at end of file diff --git a/blah2.txt.p b/blah2.txt.p new file mode 100644 index 0000000..1cc9408 --- /dev/null +++ b/blah2.txt.p @@ -0,0 +1,2 @@ +I2 +. \ No newline at end of file diff --git a/counter.py b/counter.py index 1e2fb56..69fba3d 100644 --- a/counter.py +++ b/counter.py @@ -29,7 +29,14 @@ def update_counter(file_name, reset=False): >>> update_counter('blah2.txt') 2 """ - pass + + if reset == True or not exists(file_name + ".p"): + counter = 1 + dump(counter, open(str(file_name) +".p", "wb")) + else: + counter = load(open(str(file_name) +".p", "rb")) + 1 + dump(counter, open(str(file_name) +".p", "wb")) + print counter if __name__ == '__main__': if len(sys.argv) < 2: