diff options
Diffstat (limited to 'Doc/includes/dbpickle.py')
-rw-r--r-- | Doc/includes/dbpickle.py | 14 |
1 files changed, 6 insertions, 8 deletions
diff --git a/Doc/includes/dbpickle.py b/Doc/includes/dbpickle.py index d2eee6c..1158744 100644 --- a/Doc/includes/dbpickle.py +++ b/Doc/includes/dbpickle.py @@ -46,7 +46,7 @@ class DBUnpickler(pickle.Unpickler): raise pickle.UnpicklingError("unsupported persistent object") -def main(verbose=True): +def main(): import io, pprint # Initialize and populate our database. @@ -68,20 +68,18 @@ def main(verbose=True): file = io.BytesIO() DBPickler(file).dump(memos) - if verbose: - print("Records to be pickled:") - pprint.pprint(memos) + print("Pickled records:") + pprint.pprint(memos) # Update a record, just for good measure. cursor.execute("UPDATE memos SET task='learn italian' WHERE key=1") - # Load the reports from the pickle data stream. + # Load the records from the pickle data stream. file.seek(0) memos = DBUnpickler(file, conn).load() - if verbose: - print("Unpickled records:") - pprint.pprint(memos) + print("Unpickled records:") + pprint.pprint(memos) if __name__ == '__main__': |