main.py 1.1 KB

123456789101112131415161718192021222324252627282930313233
  1. import os
  2. import psycopg2
  3. # from pprint import pprint
  4. from dotenv import load_dotenv
  5. import model
  6. import view
  7. if __name__ == '__main__':
  8. load_dotenv(verbose=True)
  9. cur = None
  10. try:
  11. conn = psycopg2.connect(user=os.getenv('db_user'),
  12. password=os.getenv('db_password'),
  13. host=os.getenv('db_host'),
  14. port=os.getenv('db_port'),
  15. database=os.getenv('db_database'))
  16. cur = conn.cursor()
  17. query = model.get_events()
  18. results = model.database_select(cur, query)
  19. broadcast = []
  20. for result in results:
  21. is_broadcast = model.is_broadcast(cur, result[0])
  22. if not is_broadcast:
  23. text = model.create_links(result[2])
  24. text = model.escape_markdown(text)
  25. print(result[1])
  26. print(text)
  27. view.broadcast(text)
  28. model.record_broadcast(cur, result[0])
  29. conn.commit()
  30. except (Exception, psycopg2.Error) as error:
  31. print("PostgreSQL error from psycopg2:", error)