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]))