Esempio n. 1
0
def main(prog, args):
    parser = argparse.ArgumentParser(prog)
    parser.add_argument('--dbdir', required=True)
    parser.add_argument('command',
                        choices=('wrap', 'versions', 'zip', 'insert',
                                 'search'))
    args, rest = parser.parse_known_args(args)
    db = wrapdb.WrapDatabase(args.dbdir)
    if args.command == 'wrap':
        print(db.get_wrap(*rest))
    elif args.command == 'versions':
        print(db.get_versions(*rest))
    elif args.command == 'latest':
        print(db.get_latest_version(*rest))
    elif args.command == 'zip':
        print(db.get_zip(*rest))
    elif args.command == 'insert':
        rest[2] = int(rest[2])
        rest[4] = rest[4].encode()
        db.insert(*rest)
    elif args.command == 'search':
        for i in db.name_search(rest[0]):
            print(i[0])
    else:
        sys.exit('Unrecognized command {!r}'.format(args.command))
Esempio n. 2
0
def _database():
    dbdir = APP.config['DB_DIRECTORY']
    return wrapdb.WrapDatabase(dbdir)
Esempio n. 3
0
 def __init__(self, dbdir='.'):
     self.dbdir = dbdir
     self.db = wrapdb.WrapDatabase(self.dbdir, True)