diff --git a/blah.txt b/blah.txt new file mode 100644 index 0000000..82dd31b --- /dev/null +++ b/blah.txt @@ -0,0 +1 @@ +€K. \ No newline at end of file diff --git a/blah2.txt b/blah2.txt new file mode 100644 index 0000000..dde5455 --- /dev/null +++ b/blah2.txt @@ -0,0 +1 @@ +€K. \ No newline at end of file diff --git a/counter.py b/counter.py index 38ec816..7c9f5ea 100644 --- a/counter.py +++ b/counter.py @@ -30,7 +30,23 @@ def update_counter(file_name, reset=False): >>> update_counter('blah2.txt') 2 """ - pass + if reset == True: + f = open(file_name,'wb') + counter = 1 + dump(counter, f) + f.close() + + elif exists(file_name) and reset == False: + f = open(file_name, 'rb+') + counter = load(f) + 1 + f.close() + + f = open(file_name,'wb') + dump(counter, f) + f.close() + + f = open(file_name,'rb+') + return load(f) if __name__ == '__main__': if len(sys.argv) < 2: