def load_order_data(filename):
headers = []
csvfile = open(filename, "r")
# Read in the headers/first row
for header in csv.DictReader(csvfile,
quotechar='"',
delimiter=',',
quoting=csv.QUOTE_ALL,
skipinitialspace=True):
headers.append(header)
for row in headers:
order = Order()
date = row['created_at']
date = datetime.strptime(date, "%Y-%m-%d %H:%M:%S.%f")
sdate = datetime.strftime(date, '%d%m%Y')
order.order_id = row['id']
order.created_at = date
order.vendor_id = row['vendor_id']
order.customer_id = row['customer_id']
try:
db.session.add(order)
db.session.commit()
except Exception as err:
db.session.rollback()
print(err)
def setUp(self):
"""
Test setup
"""
self.order = Order("order_number", "order_description", "order_price",
"size")
self.orders_list = CustomerOrders()
def get_or_create_order(update, context):
user_data = context.user_data
if 'order_id' in user_data:
order_id = user_data['order_id']
order = Order.query.filter(Order.id == order_id).first()
else:
order = Order(telegram_id=update.message.from_user.id)
db.session.add(order)
db.session.commit()
user_data['order_id'] = order.id
return order
