def getById(type, id):
try:
o = db_session.query(type).filter_by(id=object.id)
except:
db_session.rollback()
finally:
closeDbSession()
def deleteById(object, item_id):
try:
db_session.query(type(object)).filter_by(id=item_id).delete()
db_session.commit()
except:
db_session.rollback()
finally:
closeDbSession()
def add(object):
try:
db_session.add(object)
db_session.commit()
except:
db_session.rollback()
finally:
closeDbSession()
def query_email(email):
try:
r = db_session.query(User).filter_by(email =email).all()
except:
db_session.rollback()
r = None
db_session.close()
return r
def update(object):
try:
objToUpdate = db_session.query(type(object)).filter_by(id=object.id)
objToUpdate.update(object.toMappedValues())
db_session.commit()
except:
db_session.rollback()
finally:
closeDbSession()
def updatedata(data, model):
try:
db_session.query(model).filter(model.id == 1).update(data)
db_session.commit()
flash("information updated successfully", "success")
except exc.SQLAlchemyError as e:
print(e)
db_session.rollback()
flash('there was an error committing your data.', 'danger')
def update_social_media(data):
try:
for i, k in enumerate(data, start=1):
db_session.query(SocialMedia).filter(SocialMedia.site_id == i). \
update({'site_link': data[k]})
db_session.commit()
flash('Restaurant social media accounts committed successfully .',
'success')
except exc.SQLAlchemyError:
db_session.rollback()
flash('there was an error committing your data.', 'danger')
def post(self):
args = self.reqparse.parse_args()
o = UserRoles(args["role_name"])
try:
db_session.add(o)
db_session.commit()
except IntegrityError as err:
db_session.rollback()
return {"DB Error: {0}".format(err)}, 500
return o.as_dict(),201
def post(self):
args = self.reqparse.parse_args()
o = SubCategory(args["name"],args["description"],args["image_url"],args["category_id"])
try:
db_session.add(o)
db_session.commit()
except IntegrityError as err:
db_session.rollback()
return {"DB Error: {0}".format(err)}, 500
return o.as_dict(),201
def post(self):
args = self.reqparse.parse_args()
o = Orders(args["table_id"],args["waiter_id"],args["chef_id"],args["order_details"],args["order_status"],args["cost"])
try:
db_session.add(o)
db_session.commit()
except IntegrityError as err:
db_session.rollback()
return {"DB Error: {0}".format(err)}, 500
#return o.as_dict(),201
return "Success"
def add_employee(data):
try:
new_employee = Employee(employee_username=data['employee_username'],
employee_email=data['employee_email'],
employee_status=data['employee_status'])
new_employee._set_password = data['employee_password']
db_session.add(new_employee)
db_session.commit()
db_session.refresh(new_employee)
flash(
'Employee {} added successfully .'.format(
data['employee_username']), 'success')
except exc.SQLAlchemyError as e:
db_session.rollback()
flash('there was an error committing your data.', 'danger')
def add_menu_item(data):
try:
new_item = MenuItemsTable(item_name=data['item_name'],
item_category=data['item_category'],
item_status=data['item_status'],
item_price=data['item_price'],
item_size=data['item_size'],
item_image=data['item_image'])
db_session.add(new_item)
db_session.commit()
db_session.refresh(new_item)
flash('Item {} added successfully .'.format(data['item_name']),
'success')
except exc.SQLAlchemyError:
db_session.rollback()
flash('there was an error committing your data.', 'danger')
def post(self):
args = self.reqparse.parse_args()
o = Category(args["name"],args["description"],args["image_url"])
try:
db_session.add(o)
db_session.commit()
except IntegrityError as err:
db_session.rollback()
return {"DB Error: {0}".format(err)}, 500
file = args['image']
if file:
extension = os.path.splitext(file.filename)[1]
filename = "uploads/categories/category_img_" + str(o.id) + extension
o.image_url = filename
file.save(os.path.join(os.getcwd(), filename))
return o.as_dict(),201
def generate_fake(count=100, **kwargs):
"""Generate a number of fake users for testing."""
from sqlalchemy.exc import IntegrityError
from random import seed, choice
from faker import Faker
fake = Faker()
seed()
for i in range(count):
u = User(first_name=fake.first_name(),
last_name=fake.last_name(),
email=fake.email(),
password='******',
verified_email=True,
role=choice([r for r in Role]),
**kwargs)
db_session.add(u)
try:
db_session.commit()
except IntegrityError:
db_session.rollback()
def few_steps():
reload(app.utilities)
customer_details_form = CustomerDetailsForm()
delivery_customer_details_form = DeliveryCustomerDetailsForm()
# checks if the customer's POST request is within restaurant's opening days/hours
within_working_day = (
(app.utilities.from_date < datetime.datetime.now().time() <
app.utilities.to_date)
and datetime.datetime.now().strftime('%A').lower()
in app.utilities.working_days)
if request.method == 'POST' and within_working_day:
order_method = request.values.get('order-method')
verified = False
if order_method == 'Pick-up' and customer_details_form.validate():
order_array = eval(request.cookies.get('order'))
customer_name = customer_details_form.customer_name.data
customer_phone = customer_details_form.customer_phone.data
session['customer_phone'] = customer_phone
customer_email = customer_details_form.customer_email.data
address_line1 = ''
address_line2 = ''
state = ''
zipcode = ''
notes = customer_details_form.notes.data
try:
app.utilities.is_number_valid(customer_phone)
customer = Customers(customer_name=customer_name,
customer_phone=customer_phone,
customer_email=customer_email,
customer_verified=verified)
db_session.add(customer)
db_session.commit()
session['customer_id'] = customer.customer_id
order = Orders(customer_id=customer.customer_id,
order_method=order_method.lower(),
datetime_made=datetime.datetime.now(),
status='pending',
notes=notes)
db_session.add(order)
db_session.commit()
for item in order_array:
item = OrderItems(order_id=order.order_id,
item_name=item['item_name'],
item_quantity=item['item_qty'],
item_size=item['item_size'],
item_price=item['item_price'])
db_session.add(item)
db_session.commit()
except exc.SQLAlchemyError as e:
flash(
'there was an error submitting your order, please try again later',
'danger')
except TwilioException as e:
flash(
'there was an error submitting your order, please include country key with your phone number and try again',
'danger')
except:
print('an error occurred :)')
else:
return redirect(url_for('.phone_validation'))
db_session.rollback()
return render_template(
'few_steps.html',
customer_details_form=customer_details_form,
delivery_customer_details_form=delivery_customer_details_form,
allow_delivery=app.utilities.delivery_allowed,
allow_pickup=app.utilities.pickup_allowed,
delivery_taxes=app.utilities.delivery_tax,
delivery_charges=app.utilities.delivery_charges,
min_amount=app.utilities.min_amount,
max_amount=app.utilities.max_amount,
)
elif order_method == 'Delivery' and delivery_customer_details_form.validate(
):
order_array = eval(request.cookies.get('order'))
customer_name = customer_details_form.customer_name.data
customer_phone = customer_details_form.customer_phone.data
session['customer_phone'] = customer_phone
customer_email = customer_details_form.customer_email.data
address_line1 = delivery_customer_details_form.address_line1.data
address_line2 = delivery_customer_details_form.address_line2.data
state = delivery_customer_details_form.state.data
zipcode = delivery_customer_details_form.zipcode.data
notes = delivery_customer_details_form.notes.data
try:
app.utilities.is_number_valid(customer_phone)
customer = Customers(customer_name=customer_name,
customer_phone=customer_phone,
customer_email=customer_email,
customer_verified=verified)
db_session.add(customer)
db_session.commit()
session['customer_id'] = customer.customer_id
order = Orders(customer_id=customer.customer_id,
order_method=order_method.lower(),
datetime_made=datetime.datetime.now(),
address_line1=address_line1,
address_line2=address_line2,
state=state,
zipcode=zipcode,
status='pending',
notes=notes)
db_session.add(order)
db_session.commit()
for item in order_array:
item = OrderItems(order_id=order.order_id,
item_name=item['item_name'],
item_quantity=item['item_qty'],
item_size=item['item_size'],
item_price=item['item_price'])
db_session.add(item)
db_session.commit()
except exc.SQLAlchemyError as e:
flash(
'there was an error submitting your order, please try again later',
'danger')
except TwilioException as e:
flash(
'there was an error submitting your order, please include country key with your phone number and try again',
'danger')
except:
print('an error occurred :)')
else:
return redirect(url_for('.phone_validation'))
db_session.rollback()
return render_template(
'few_steps.html',
customer_details_form=customer_details_form,
delivery_customer_details_form=delivery_customer_details_form,
allow_delivery=app.utilities.delivery_allowed,
allow_pickup=app.utilities.pickup_allowed,
delivery_taxes=app.utilities.delivery_tax,
delivery_charges=app.utilities.delivery_charges,
min_amount=app.utilities.min_amount,
max_amount=app.utilities.max_amount)
return render_template(
'few_steps.html',
customer_details_form=customer_details_form,
delivery_customer_details_form=delivery_customer_details_form,
allow_delivery=app.utilities.delivery_allowed,
allow_pickup=app.utilities.pickup_allowed,
delivery_taxes=app.utilities.delivery_tax,
delivery_charges=app.utilities.delivery_charges,
min_amount=app.utilities.min_amount,
max_amount=app.utilities.max_amount,
)
elif request.method == 'POST' and within_working_day is False:
flash(
'You can place orders within working hours/days: {}, From {} To {}.'
.format(', '.join(app.utilities.working_days).title(),
app.utilities.from_date.strftime('%H:%M %p'),
app.utilities.to_date.strftime('%H:%M %p')), 'warning')
return render_template(
'few_steps.html',
customer_details_form=customer_details_form,
delivery_customer_details_form=delivery_customer_details_form,
allow_pickup=app.utilities.pickup_allowed,
allow_delivery=app.utilities.delivery_allowed,
delivery_taxes=app.utilities.delivery_tax,
delivery_charges=app.utilities.delivery_charges,
min_amount=app.utilities.min_amount,
max_amount=app.utilities.max_amount,
)
return redirect(
url_for('.menu',
allow_delivery=app.utilities.delivery_allowed,
delivery_taxes=app.utilities.delivery_tax,
delivery_charges=app.utilities.delivery_charges,
min_amount=app.utilities.min_amount,
max_amount=app.utilities.max_amount))
def process_components(sim_path):
# dump components
comps = re.findall(r'filename:[ \t]*([^\s]+)',
file(sim_path).read())
file(dirname(sim_path) + "/comps.json", "w").write(json.dumps(comps))
# plot components
for comp in comps:
for mode in ("lin", "log"):
plot(dirname(sim_path) + '/' + comp,
outfile=dirname(sim_path) + ('/plot-%s-%s.gif' % (comp, mode)),
logger=appendLog,
logy=(mode == "log"))
os.path.walk(workdir % 'mcstas',
lambda _arg, folder, files:
process_components(folder + "/mcstas.sim"),
[])
if __name__ == '__main__':
while True:
time.sleep(1)
try:
work()
except:
# rollback any non-committed transactions
db_session.rollback()
# print error message to stdout
print 'Job failed because:'
traceback.print_exc()
def process_components(sim_path):
# dump components
comps = re.findall(r'filename:[ \t]*([^\s]+)', file(sim_path).read())
file(dirname(sim_path) + "/comps.json", "w").write(json.dumps(comps))
# plot components
for comp in comps:
for mode in ("lin", "log"):
plot(dirname(sim_path) + '/' + comp,
outfile=dirname(sim_path) + ('/plot-%s-%s.gif' %
(comp, mode)),
logger=appendLog,
logy=(mode == "log"))
os.path.walk(
workdir % 'mcstas',
lambda _arg, folder, files: process_components(folder + "/mcstas.sim"),
[])
if __name__ == '__main__':
while True:
time.sleep(1)
try:
work()
except:
# rollback any non-committed transactions
db_session.rollback()
# print error message to stdout
print 'Job failed because:'
traceback.print_exc()
