From 7bc07dc2f50e00653c33f459277aa47a50b6ab1c Mon Sep 17 00:00:00 2001 From: Halliax Date: Mon, 14 Mar 2016 01:57:51 -0400 Subject: [PATCH] This is done. It pickles correctly now. --- counter.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/counter.py b/counter.py index 1e2fb56..762bdcc 100644 --- a/counter.py +++ b/counter.py @@ -29,11 +29,24 @@ def update_counter(file_name, reset=False): >>> update_counter('blah2.txt') 2 """ - pass + if exists(file_name) and not reset: + fin = open(file_name,'r+') + loaded = load(fin) + loaded += 1 + res = loaded + fin.seek(0,0) + dump(loaded,fin) + fin.close() + else: + fin = open(file_name,'w') + dump(1,fin) + res = 1 + fin.close() + return res if __name__ == '__main__': if len(sys.argv) < 2: import doctest doctest.testmod() else: - print "new value is " + str(update_counter(sys.argv[1])) \ No newline at end of file + print "new value is " + str(update_counter(sys.argv[1]))