Skip to content

louise submits pickling toolbox #55

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
nielseneli wants to merge 1 commit into
base: master
Choose a base branch
from
Open

louise submits pickling toolbox #55

Show file tree
Hide file tree
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
2 changes: 2 additions & 0 deletions blah.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
I3
.
2 changes: 2 additions & 0 deletions blah2.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
I2
.
82 changes: 51 additions & 31 deletions counter.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,39 +1,59 @@
""" A program that stores and updates a counter using a Python pickle file"""
"""A program that stores and updates a counter using a Python pickle file.
"""

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


def update_counter(file_name, 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
the reset flag is True.

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
doesn't exist, a counter is created and initialized to 1.
reset: True if the counter in the file should be rest.
returns: the new counter value

>>> update_counter('blah.txt',True)
1
>>> update_counter('blah.txt')
2
>>> update_counter('blah2.txt',True)
1
>>> update_counter('blah.txt')
3
>>> update_counter('blah2.txt')
2
"""
pass
"""Updates a counter stored in the file 'file_name.'
A new counter will be created and initialized to 1 if none exists
or if the reset flag is True.

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 doesn't exist, a counter is created and initialized
to 1.
reset: True if the counter in the file should be rest.
returns: the new counter value

>>> update_counter('blah.txt',True)
1
>>> update_counter('blah.txt')
2
>>> update_counter('blah2.txt',True)
1
>>> update_counter('blah.txt')
3
>>> update_counter('blah2.txt')
2
"""
# check if the file exists
extant = exists(file_name)
# if the file exists and is not supposed to be reset, this opens the file,
# increments the counter by one, and closes the file
if extant and not reset:
f = open(file_name, 'r+')
f.seek(0, 0)
count = load(f)
count += 1
# otherwise, this opens the file and initializes the counter at 1.
else:
f = open(file_name, 'w')
f.seek(0, 0)
count = 1
f.seek(0, 0)
dump(count, f)
f.close()
return count


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