database.py

#!/usr/local/bin/python
# coding: ascii

import sqlite3
import argparse
import os.path

def init_db(dbcon):
	f = open(os.path.join(os.path.dirname(__file__),'avatar-schema.sql'),'r')
	sql = f.read()
	dbcon.executescript(sql)

def dump_db(dbcon):
	with open(os.path.join(os.path.dirname(__file__),'avatar-dump.sql'), 'w') as f:
		for line in dbcon.iterdump():
			f.write('%s\n' % line)

def connect_db():
	dbcon = None

	# check if db file exists
	if (not os.path.isfile(os.path.join(os.path.dirname(__file__),'avatar.db'))):
		dbcon = sqlite3.connect(os.path.join(os.path.dirname(__file__),'avatar.db'), detect_types=sqlite3.PARSE_DECLTYPES)
		init_db(dbcon)
	else:
		dbcon = sqlite3.connect(os.path.join(os.path.dirname(__file__),'avatar.db'), detect_types=sqlite3.PARSE_DECLTYPES)

	return dbcon

if __name__ == "__main__":
	parser = argparse.ArgumentParser(description='Database')
	parser.add_argument('--init', help='Init the database', action="store_true")
	parser.add_argument('--dump', help='Dump the database', action="store_true")
	args = parser.parse_args()

	dbcon = None

	try:
		dbcon = connect_db()

		if (args.init):
			init_db(dbcon)
		elif(args.dump):
			dump_db(dbcon)

	finally:
		if dbcon:
			dbcon.close()