Skip to content

Toolbox submission #35

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
gabclarke wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Toolbox submission #35

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
37 changes: 26 additions & 11 deletions counter.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,9 @@

from os.path import exists
import sys
from pickle import dump, load
import pickle

def update_counter(file_name, reset=False):
def updateCounter(fileName, reset=False):
""" Updates a counter stored in the file 'file_name'

A new counter will be created and initialized to 1 if none exists or if
Expand All @@ -13,27 +13,42 @@ def update_counter(file_name, reset=False):
If the counter already exists and reset is False, the counter's value will
be incremented.

file_name: the file that stores the counter to be incremented. If the file
fileName: the file that stores the counter to be incremented. If the file
doesn't exist, a counter is created and initialized to 1.
reset: True if the counter in the file should be rest.
reset: True if the counter in the file should be reset.
returns: the new counter value

>>> update_counter('blah.txt',True)
>>> updateCounter('blah.txt',True)
1
>>> update_counter('blah.txt')
>>> updateCounter('blah.txt')
2
>>> update_counter('blah2.txt',True)
>>> updateCounter('blah2.txt',True)
1
>>> update_counter('blah.txt')
>>> updateCounter('blah.txt')
3
>>> update_counter('blah2.txt')
>>> updateCounter('blah2.txt')
2
"""
pass
if exists(fileName) and not reset:
f = open(fileName, 'r+')
content = int(pickle.load(f))
content += 1
f.seek(0,0)
pickle.dump(content, f)
f.close()
return content

else:
f = open(fileName, 'w')
pickle.dump('1', f)
f.seek(0,0)
f.close
return 1


if __name__ == '__main__':
if len(sys.argv) < 2:
import doctest
doctest.testmod()
else:
print "new value is " + str(update_counter(sys.argv[1]))
print "new value is " + str(updateCounter(sys.argv[1]))