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