From bff6af28bd7fe90cf934f222d1ee349a24655801 Mon Sep 17 00:00:00 2001
From: KatieButler <katiethetraveller@gmail.com>
Date: Sun, 13 Mar 2016 17:18:36 -0400
Subject: [PATCH] turning in Pickling ToolBox

---
 counter.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/counter.py b/counter.py
index 1e2fb56..e8b93ac 100644
--- a/counter.py
+++ b/counter.py
@@ -29,11 +29,19 @@ def update_counter(file_name, reset=False):
 	>>> update_counter('blah2.txt')
 	2
 	"""
-	pass
+	counter = 0
+	if reset == True or not exists(file_name):
+		counter = 1
+		dump(counter, open(file_name, 'w'))
+	else:
+		counter = load(open(file_name, 'r')) + 1
+		dump(counter, open(file_name, 'w'))
+	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]))