45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
|
#!/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()
|