diff options
| -rw-r--r-- | db_create.py | 98 |
1 files changed, 98 insertions, 0 deletions
diff --git a/db_create.py b/db_create.py new file mode 100644 index 00000000..c6eaec7d --- /dev/null +++ b/db_create.py @@ -0,0 +1,98 @@ +#! /usr/bin/env python +# -*- coding: utf-8 -*- + +# Grenouille - An online service for weather data. +# Copyright (C) 2014 Cédric Bonhomme - http://cedricbonhomme.org/ +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as +# published by the Free Software Foundation, either version 3 of the +# License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +__author__ = "Cedric Bonhomme" +__version__ = "$Revision: 0.2 $" +__date__ = "$Date: 2014/03/16 $" +__revision__ = "$Date: 2014/03/24 $" +__copyright__ = "Copyright (c) Cedric Bonhomme" +__license__ = "AGPLv3" + +from pyaggr3g470r import db +from pyaggr3g470r.models import User, Feed, Role +from werkzeug import generate_password_hash + +from sqlalchemy.engine import reflection +from sqlalchemy.schema import ( + MetaData, + Table, + DropTable, + ForeignKeyConstraint, + DropConstraint, + ) + +def db_DropEverything(db): + # From http://www.sqlalchemy.org/trac/wiki/UsageRecipes/DropEverything + + conn=db.engine.connect() + + # the transaction only applies if the DB supports + # transactional DDL, i.e. Postgresql, MS SQL Server + trans = conn.begin() + + inspector = reflection.Inspector.from_engine(db.engine) + + # gather all data first before dropping anything. + # some DBs lock after things have been dropped in + # a transaction. + metadata = MetaData() + + tbs = [] + all_fks = [] + + for table_name in inspector.get_table_names(): + fks = [] + for fk in inspector.get_foreign_keys(table_name): + if not fk['name']: + continue + fks.append( + ForeignKeyConstraint((),(),name=fk['name']) + ) + t = Table(table_name,metadata,*fks) + tbs.append(t) + all_fks.extend(fks) + + for fkc in all_fks: + conn.execute(DropConstraint(fkc)) + + for table in tbs: + conn.execute(DropTable(table)) + + trans.commit() + + + +db_DropEverything(db) +db.create_all() + +role_admin = Role(name="admin") +role_user = Role(name="user") + +user1 = User(firstname="admin", lastname="admin", email="root@pyAggr3g470r.localhost", pwdhash=generate_password_hash("root")) + +user1.roles.extend([role_admin, role_user]) + +feed1 = Feed(title="Armed and Dangerous", description="Sex, software, politics, and firearms. Life's simple pleasures...", + link="http://esr.ibiblio.org/?feed=rss2", site_link="http://esr.ibiblio.org/", + email_notification=False, enabled=True, user_id=user1.id) + +user1.feeds.extend([feed1]) + +db.session.add(user1) +db.session.commit()
\ No newline at end of file |