Forgot to push this little guy
Good thing I cleaned up my working directory. It only exists because pickles created from database.py itself won't be openable from a file simply importing databse.py. So we create it when in 'imported state'.
This commit is contained in:
parent
7851b038f5
commit
ea0855bd00
44
db.py
Executable file
44
db.py
Executable file
|
@ -0,0 +1,44 @@
|
||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
import argparse
|
||||||
|
import database
|
||||||
|
import time
|
||||||
|
import os
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
|
||||||
|
# Parsing arguments
|
||||||
|
parser = argparse.ArgumentParser(
|
||||||
|
description="Database operations")
|
||||||
|
parser.add_argument(
|
||||||
|
'-i', '--initialize', action='store_true',
|
||||||
|
help="Reconstruct the whole database")
|
||||||
|
parser.add_argument(
|
||||||
|
'-p', '--prune', action='store_true',
|
||||||
|
help="Remove old entries from database")
|
||||||
|
parser.add_argument(
|
||||||
|
'-b', '--prune-base', action='store_true',
|
||||||
|
help="TODO")
|
||||||
|
parser.add_argument(
|
||||||
|
'-s', '--prune-before', type=int,
|
||||||
|
default=(int(time.time()) - 60*60*24*31*6),
|
||||||
|
help="TODO")
|
||||||
|
parser.add_argument(
|
||||||
|
'-r', '--references', action='store_true',
|
||||||
|
help="Update the reference count")
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
if not args.initialize:
|
||||||
|
DB = database.Database()
|
||||||
|
else:
|
||||||
|
if os.path.isfile(database.Database.PATH):
|
||||||
|
os.unlink(database.Database.PATH)
|
||||||
|
DB = database.Database()
|
||||||
|
|
||||||
|
DB.enter_step('main')
|
||||||
|
if args.prune:
|
||||||
|
DB.prune(before=args.prune_before, base_only=args.prune_base)
|
||||||
|
if args.references:
|
||||||
|
DB.update_references()
|
||||||
|
|
||||||
|
DB.save()
|
Loading…
Reference in a new issue