##// END OF EJS Templates
db manage added more logging, set custom logger and add optional print sql statments
db manage added more logging, set custom logger and add optional print sql statments

File last commit:

r229:69a29242 default
r229:69a29242 default
Show More
db_manage.py
74 lines | 2.3 KiB | text/x-python | PythonLexer
Marcin Kuzminski
added db_manage script
r59 import logging
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226 from os.path import dirname as dn
from sqlalchemy.engine import create_engine
Marcin Kuzminski
added db_manage script
r59 import os
fixed dbmanage script to work as standalone script
r228 import sys
ROOT = dn(dn(dn(os.path.realpath(__file__))))
sys.path.append(ROOT)
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226 from pylons_app.model.db import Users
db manage added more logging, set custom logger and add optional print sql statments
r229 from pylons_app.model.meta import Session, Base
Marcin Kuzminski
added db_manage script
r59
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226 from pylons_app.lib.auth import get_crypt_password
from pylons_app.model import init_model
Marcin Kuzminski
added db_manage script
r59
db manage added more logging, set custom logger and add optional print sql statments
r229 log = logging.getLogger('db manage')
log.setLevel(logging.DEBUG)
console_handler = logging.StreamHandler()
console_handler.setFormatter(logging.Formatter("%(asctime)s.%(msecs)03d"
" %(levelname)-5.5s [%(name)s] %(message)s"))
log.addHandler(console_handler)
Marcin Kuzminski
updated db manage script for creating interactive admin account and db
r66
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226 class DbManage(object):
db manage added more logging, set custom logger and add optional print sql statments
r229 def __init__(self, log_sql):
self.dbname = 'hg_app.db'
dburi = 'sqlite:////%s' % os.path.join(ROOT, self.dbname)
engine = create_engine(dburi, echo=log_sql)
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226 init_model(engine)
self.sa = Session()
def check_for_db(self, override):
db manage added more logging, set custom logger and add optional print sql statments
r229 log.info('checking for exisiting db')
if os.path.isfile(os.path.join(ROOT, self.dbname)):
log.info('database exisist')
if not override:
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226 raise Exception('database already exists')
db manage added more logging, set custom logger and add optional print sql statments
r229
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226 def create_tables(self, override=False):
"""
Create a auth database
"""
self.check_for_db(override)
db manage added more logging, set custom logger and add optional print sql statments
r229 if override:
log.info("database exisist and it's going to be destroyed")
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226 Base.metadata.create_all(checkfirst=override)
db manage added more logging, set custom logger and add optional print sql statments
r229 log.info('Created tables for %s', self.dbname)
Marcin Kuzminski
added db_manage script
r59
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226 def admin_prompt(self):
import getpass
db manage added more logging, set custom logger and add optional print sql statments
r229 username = raw_input('Specify admin username:')
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226 password = getpass.getpass('Specify admin password:')
self.create_user(username, password, True)
def create_user(self, username, password, admin=False):
db manage added more logging, set custom logger and add optional print sql statments
r229 log.info('creating administrator user %s', username)
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226
new_user = Users()
new_user.username = username
new_user.password = get_crypt_password(password)
new_user.admin = admin
new_user.active = True
try:
self.sa.add(new_user)
self.sa.commit()
except:
self.sa.rollback()
raise
Marcin Kuzminski
updated db manage script for creating interactive admin account and db
r66
if __name__ == '__main__':
db manage added more logging, set custom logger and add optional print sql statments
r229 dbmanage = DbManage(log_sql=True)
rewritten db manage script to use sqlalchemy. Fixed sqlalchemy models to more generic.
r226 dbmanage.create_tables(override=True)
dbmanage.admin_prompt()
Marcin Kuzminski
updated db manage script for creating interactive admin account and db
r66