helpers.py 690 B

1234567891011121314151617181920212223242526272829303132
  1. import jinja2
  2. from pathlib import Path
  3. from collections import namedtuple
  4. import random
  5. def sanitize(query):
  6. if not query.isnumeric():
  7. res = 1
  8. else:
  9. query = int(query)
  10. if query < 1:
  11. res = 1
  12. elif query > 500:
  13. res = 500
  14. else:
  15. res = query
  16. return res
  17. def generate_ids(num_queries):
  18. return random.sample(range(1, 10000), num_queries)
  19. def load_template():
  20. path = Path("templates", "fortune.html")
  21. with open(str(path), "r") as template_file:
  22. template_text = template_file.read()
  23. return jinja2.Template(template_text)
  24. FortuneTuple = namedtuple("Fortune", ["id", "message"])