kallithea Changeset - 911dab498eb2

Changeset - 911dab498eb2

Parent rev.

Child rev.

[Not reviewed]

default

0 1 0

Marcin Kuzminski - 16 years ago 2010-04-19 22:52:31
marcin@python-works.com

Updated db manage

1 file changed with 7 insertions and 6 deletions:

pylons_app/lib/db_manage.py

0 comments (0 inline, 0 general)

pylons_app/lib/db_manage.py

➞

Show inline comments

@@ @@ -9,21 +9,22 @@ logging.basicConfig(level=logging.DEBUG) @@
 def get_sqlite_conn_cur():
     conn = sqlite3.connect(os.path.join(ROOT, 'hg_app.db'))
     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')
         cur.execute("""DROP TABLE IF EXISTS users """)
         cur.execute("""CREATE TABLE users
                         (user_id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ @@ -62,10 +63,10 @@ def create_user(username, password, admi @@
         conn.commit()
     except:
         conn.rollback()
         raise
 if __name__ == '__main__':
     create_tables()
+    create_tables(True)
     admin_prompt()

0 comments (0 inline, 0 general)