diff --git a/blah.txt b/blah.txt new file mode 100644 index 0000000..82dd31b --- /dev/null +++ b/blah.txt @@ -0,0 +1 @@ +€K. \ No newline at end of file diff --git a/blah2.txt b/blah2.txt new file mode 100644 index 0000000..dde5455 --- /dev/null +++ b/blah2.txt @@ -0,0 +1 @@ +€K. \ No newline at end of file diff --git a/counter.py b/counter.py index 38ec816..607c1de 100644 --- a/counter.py +++ b/counter.py @@ -4,7 +4,6 @@ import sys from pickle import dump, load - def update_counter(file_name, reset=False): """ Updates a counter stored in the file 'file_name' @@ -30,7 +29,21 @@ def update_counter(file_name, reset=False): >>> update_counter('blah2.txt') 2 """ - pass + count = 0 + + if reset==False: + if exists( file_name ): + count = int(load( open( file_name, 'rb' ) ) ) + + elif reset==True: + pass + + else: + raise TypeError('Boolean expected') + + count = count + 1 + dump( count, open( file_name, 'wb' ) ) + return count if __name__ == '__main__': if len(sys.argv) < 2: diff --git a/var.pickle b/var.pickle new file mode 100644 index 0000000..82dd31b --- /dev/null +++ b/var.pickle @@ -0,0 +1 @@ +€K. \ No newline at end of file