diff --git a/blah.p b/blah.p new file mode 100644 index 0000000..7621841 Binary files /dev/null and b/blah.p differ diff --git a/blah.txt b/blah.txt new file mode 100644 index 0000000..7621841 Binary files /dev/null and b/blah.txt differ diff --git a/blah2.p b/blah2.p new file mode 100644 index 0000000..04ffe6b Binary files /dev/null and b/blah2.p differ diff --git a/blah2.txt b/blah2.txt new file mode 100644 index 0000000..04ffe6b Binary files /dev/null and b/blah2.txt differ diff --git a/counter.py b/counter.py index 38ec816..98ba282 100644 --- a/counter.py +++ b/counter.py @@ -2,7 +2,7 @@ from os.path import exists import sys -from pickle import dump, load +import pickle def update_counter(file_name, reset=False): @@ -19,18 +19,31 @@ def update_counter(file_name, reset=False): reset: True if the counter in the file should be rest. returns: the new counter value - >>> update_counter('blah.txt',True) + >>> update_counter('blah.p',True) 1 - >>> update_counter('blah.txt') + >>> update_counter('blah.p') 2 - >>> update_counter('blah2.txt',True) + >>> update_counter('blah2.p',True) 1 - >>> update_counter('blah.txt') + >>> update_counter('blah.p') 3 - >>> update_counter('blah2.txt') + >>> update_counter('blah2.p') 2 """ - pass + if not reset: + try: + counter = int(pickle.load(open(file_name, "rb"))) + counter += 1 + print(counter) + pickle.dump(str(counter), open(file_name, "wb")) + except FileNotFoundError: + open(file_name, 'a').close() + pickle.dump(str(1), open(file_name, "wb")) + print(1) + else: + open(file_name, 'a').close() + pickle.dump(str(1), open(file_name, "wb")) + print(1) if __name__ == '__main__': if len(sys.argv) < 2: diff --git a/jeff.p b/jeff.p new file mode 100644 index 0000000..9a29eb5 Binary files /dev/null and b/jeff.p differ diff --git a/jeff.txt b/jeff.txt new file mode 100644 index 0000000..e69de29