changeset 88:911dab498eb2

Updated db manage
author Marcin Kuzminski <marcin@python-works.com>
date Mon, 19 Apr 2010 22:52:31 +0200
parents 9f6300b96380
children b2c38dee135a
files pylons_app/lib/db_manage.py
diffstat 1 files changed, 7 insertions(+), 6 deletions(-) [+]
line wrap: on
line diff
--- a/pylons_app/lib/db_manage.py	Sun Apr 18 21:15:53 2010 +0200
+++ b/pylons_app/lib/db_manage.py	Mon Apr 19 22:52:31 2010 +0200
@@ -12,15 +12,16 @@
     cur = conn.cursor()
     return conn, cur
 
-def check_for_db():
-    if os.path.isfile(os.path.join(ROOT, 'hg_app.db')):
-        raise Exception('database already exists')
+def check_for_db(override):
+    if not override:
+        if os.path.isfile(os.path.join(ROOT, 'hg_app.db')):
+            raise Exception('database already exists')
 
-def create_tables():
+def create_tables(override=False):
     """
     Create a auth database
     """
-    check_for_db()
+    check_for_db(override)
     conn, cur = get_sqlite_conn_cur()
     try:
         logging.info('creating table %s', 'users')
@@ -65,7 +66,7 @@
         raise
     
 if __name__ == '__main__':
-    create_tables()
+    create_tables(True)
     admin_prompt()