From 9e38e877f72b1a45538278b181a1382a3bd89df6 Mon Sep 17 00:00:00 2001 From: jaredbriskman Date: Sun, 13 Mar 2016 20:20:18 -0400 Subject: [PATCH 1/2] Finished toolbox! --- counter.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/counter.py b/counter.py index 1e2fb56..21567bb 100644 --- a/counter.py +++ b/counter.py @@ -29,11 +29,23 @@ def update_counter(file_name, reset=False): >>> update_counter('blah2.txt') 2 """ - pass + if not exists(file_name) or reset: + with open(file_name, 'w') as f: + dump(1,f) + f.close() + return 1 + else: + with open(file_name, 'r+') as f: + counter = load(f) + 1 #incremented + f.seek(0,0) + dump(counter,f) + f.close() + return counter + 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])) From d52c8af1fba56fef87f19a0d7300432818feea89 Mon Sep 17 00:00:00 2001 From: jaredbriskman Date: Sun, 13 Mar 2016 20:22:24 -0400 Subject: [PATCH 2/2] More pythonic and pithy --- counter.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/counter.py b/counter.py index 21567bb..aa01a40 100644 --- a/counter.py +++ b/counter.py @@ -32,15 +32,13 @@ def update_counter(file_name, reset=False): if not exists(file_name) or reset: with open(file_name, 'w') as f: dump(1,f) - f.close() - return 1 + return 1 else: with open(file_name, 'r+') as f: counter = load(f) + 1 #incremented f.seek(0,0) dump(counter,f) - f.close() - return counter + return counter if __name__ == '__main__':