def add_iata_to_db(iata): if db_session.query(IATA).filter_by(code=iata).count() == 0: iata_model = IATA(code=iata) db_session.add(iata_model) db_session.commit() return iata_model else: return db_session.query(IATA).filter_by(code=iata)[0]
def create(args: CreateParams): hotels = Hotels(name=args.name, address=args.address, created_at=datetime.now(), updated_at=datetime.now()) db_session.add(hotels) db_session.commit() db_session.refresh(hotels) return hotels
def queue_partner_emails(): try: users = db_session.query(Users).filter( Users.wants_emails == True).all() for user in users: print("Sending email to {}...".format(user.email)) partner = get_next_partner(user.uuid) if not partner: print("Reseting prayers for {}!".format(user.email)) db_session.query(Prayers).filter( Prayers.user_id == str(user.uuid)).filter( Prayers.partner_id != None).delete( synchronize_session=False) db_session.commit() partner = get_next_partner(user.uuid) message = Mail(from_email=From('*****@*****.**', '3ABC Prayer Reminders'), to_emails=user.email) message.dynamic_template_data = { 'user_name': user.name, 'email': partner.email, 'name': partner.name, 'type': partner.type, 'image': partner.image_url if partner.image_url is not None else None, 'description': partner.description if partner.description is not None else None, 'location': partner.location if partner.location is not None else None, } message.template_id = 'd-5f8116bd407849d5a06e66d586354bdb' sg = SendGridAPIClient( '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' ) response = sg.send(message) print(response.status_code) new_prayer = Prayers(member_id=None, partner_id=partner.ccbid, user_id=user.uuid) db_session.add(new_prayer) db_session.commit() return 200 except Exception as e: print('Failed sending partner emails') print(e) return 500 finally: db_session.close()
def add_content_server(hostname, name, host_url): """Add a new content server to the database""" content_server = ContentServers(hostname=hostname, name=name, host_url=host_url) db_session.add(content_server) db_session.commit() click.echo(f'Content server added\n' f'hostname: {hostname}\nname:{name}\nurl:{host_url}')
def create_monitor(): """generate monitor data""" # logger logger = logging.getLogger("monitors") codes = [200, 200, 200, 500, 404] for i in range(5): mon = MonitorFactory( endpoint=f"http://apidash.dev:8000/api/v1/status/{codes[i]}") logger.info(f"Adding monitor {mon.name}") db_session.add(mon) db_session.commit()
def queue_emails(): try: users = db_session.query(Users).filter( Users.wants_emails == True).all() for user in users: print("Sending email to {}...".format(user.email)) member = get_next_member(user.uuid) if not member: print("Reseting prayers for {}!".format(user.email)) db_session.query(Prayers).filter( Prayers.user_id == str(user.uuid)).filter( Prayers.member_id != None).delete( synchronize_session=False) db_session.commit() member = get_next_member(user.uuid) message = Mail(from_email=From('*****@*****.**', '3ABC Prayer Reminders'), to_emails=user.email) member_since = member.member_since.strftime('%B %Y') message.dynamic_template_data = { 'email': member.email, 'name': member.name, 'member_since': member_since, 'image': member.image_url if member.image_url is not None else None, 'username': user.name if user.name is not None else user.username } message.template_id = 'd-0a35a380ad214eb0b6152ddaa961f420' sg = SendGridAPIClient( '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' ) response = sg.send(message) print(response.status_code) new_prayer = Prayers(member_id=member.ccbid, partner_id=None, user_id=user.uuid) db_session.add(new_prayer) db_session.commit() return 200 except Exception as e: print('Failed sending all emails') print(e) return 500 finally: db_session.close()
def add_ticket_to_db(ticket, date_search, iata_from_model, iata_to_model): ticket_info = check_ticket(ticket) if ticket_info["flights"][0]["invalid"] == 1: return ticket["price"] = ticket_info["flights_price"] new_ticket = Ticket( fly_from=iata_from_model, fly_to=iata_to_model, date_from=ticket["date_from"], date_to=ticket["date_to"], date_search=date_search, booking_token=ticket["booking_token"], price=ticket["price"], ) db_session.add(new_ticket) db_session.commit()