fixing the db script

This commit is contained in:
Feliciaan De Palmenaer 2015-04-03 18:11:39 +02:00
parent 8a975d5d44
commit e1b35de786
3 changed files with 67 additions and 34 deletions

View file

@ -2,6 +2,8 @@ from models import *
from app import db from app import db
from itertools import product from itertools import product
def recreate_db():
db.drop_all() db.drop_all()
db.create_all() db.create_all()
@ -28,6 +30,8 @@ sauzen = ["Balisaus", "Yu siang saus", "Gon boa saus", "Curry saus", "Oestersaus
specials = ["Kippenbolletjes zoetzuur", "varkenbolletjes zoetzuur", "Nazi Babi Pangang", "Bami Babi Pangang", specials = ["Kippenbolletjes zoetzuur", "varkenbolletjes zoetzuur", "Nazi Babi Pangang", "Bami Babi Pangang",
"Diverse groenten met bami(Vegetarisch)", "Diverse groenten met nazi(Vegetarisch)"] "Diverse groenten met bami(Vegetarisch)", "Diverse groenten met nazi(Vegetarisch)"]
def add_chinees():
chinees = Location() chinees = Location()
chinees.configure("Oceans's Garden", "Zwijnaardsesteenweg 399 9000 Gent, tel: 09/222.72.74", "http://oceangarden.byethost3.com/studentenmenus.html") chinees.configure("Oceans's Garden", "Zwijnaardsesteenweg 399 9000 Gent, tel: 09/222.72.74", "http://oceangarden.byethost3.com/studentenmenus.html")
db.session.add(chinees) db.session.add(chinees)
@ -37,12 +41,12 @@ def chinees_create_entry(zetmeel, vlees="", saus=""):
entry.configure(chinees, "{} {} {}".format(zetmeel, vlees, saus).rstrip(), 550) entry.configure(chinees, "{} {} {}".format(zetmeel, vlees, saus).rstrip(), 550)
db.session.add(entry) db.session.add(entry)
for zetmeel, vlees, saus in product(zetmelen, vlezen, sauzen): for zetmeel, vlees, saus in product(zetmelen, vlezen, sauzen):
chinees_create_entry(zetmeel, vlees, saus) chinees_create_entry(zetmeel, vlees, saus)
for special in specials: for special in specials:
chinees_create_entry(special) chinees_create_entry(special)
############################################# #############################################
############################################ ############################################
@ -53,6 +57,8 @@ pizzas = ['Bolognese de luxe', 'Hawaï', 'Popeye', 'Pepperoni', 'Seafood', 'Hot
'Green fiësta', 'Chicken bbq', 'Funky chicken', 'Veggie', 'Meat lovers', 'Scampi mampi', 'Tabasco', 'Green fiësta', 'Chicken bbq', 'Funky chicken', 'Veggie', 'Meat lovers', 'Scampi mampi', 'Tabasco',
'Chicken time', 'Meatballs', 'Tuna', 'Anchovy', 'Calzone', 'Bbq meatballs', 'Creamy chicken', 'Hot bolognese'] 'Chicken time', 'Meatballs', 'Tuna', 'Anchovy', 'Calzone', 'Bbq meatballs', 'Creamy chicken', 'Hot bolognese']
def add_simpizza():
simpizza = Location() simpizza = Location()
simpizza.configure("Sim-pizza", "De Pintelaan 252 9000 Gent, tel: 09/321.02.00", "http://simpizza.be") simpizza.configure("Sim-pizza", "De Pintelaan 252 9000 Gent, tel: 09/321.02.00", "http://simpizza.be")
db.session.add(simpizza) db.session.add(simpizza)
@ -62,5 +68,31 @@ for pizza in pizzas:
entry.configure(simpizza, pizza, 1195) entry.configure(simpizza, pizza, 1195)
db.session.add(entry) db.session.add(entry)
def commit():
# commit all the things # commit all the things
db.session.commit() db.session.commit()
print("Committing successful")
locations = {'Ocean\'s Garden': add_chinees, 'Sim-pizza': add_simpizza}
print('Create database script!')
print('=======================\n\n')
db_key = 'recreate this database now!'
input_db = input("To recreate the db type: '" + db_key + "'")
if db_key in input_db:
print("recreating the database!")
recreate_db()
for location in locations.values():
location()
commit()
else:
print("not recreating the database!")
# DID NOT RECREATE ASK IF YOU WANT TO RECREATE THE NEXT ONES
for name, function in locations.items():
add_location = input("Do you want to add " + name + "? Press: 'Y'")
print("Adding " + name)
if add_location.lower() in ['y', 'yes']:
function()
commit()

View file

@ -7,7 +7,7 @@ from app import db
# Create database models # Create database models
class User(db.Model): class User(db.Model):
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), unique=True) username = db.Column(db.String(80), unique=True, nullable=False)
admin = db.Column(db.Boolean) admin = db.Column(db.Boolean)
bias = db.Column(db.Integer) bias = db.Column(db.Integer)
runs = db.relationship('Order', backref='courrier', lazy='dynamic') runs = db.relationship('Order', backref='courrier', lazy='dynamic')
@ -124,7 +124,7 @@ class Order(db.Model):
class OrderItem(db.Model): class OrderItem(db.Model):
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id')) user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
order_id = db.Column(db.Integer, db.ForeignKey('order.id')) order_id = db.Column(db.Integer, db.ForeignKey('order.id'), nullable=False)
product_id = db.Column(db.Integer, db.ForeignKey('product.id')) product_id = db.Column(db.Integer, db.ForeignKey('product.id'))
name = db.Column(db.String(120)) name = db.Column(db.String(120))

View file

@ -35,6 +35,7 @@ def handle404(e):
def handle401(e): def handle401(e):
return render_template('errors/401.html'), 401 return render_template('errors/401.html'), 401
class AnonymouseUser: class AnonymouseUser:
def is_active(self): def is_active(self):
return False return False