utils.py 6.2KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127
  1. # -*- coding: utf-8 -*-
  2. import datetime
  3. import random
  4. import string
  5. from enum import Enum
  6. from redis import Redis
  7. from rq import Queue
  8. DATETIME_FORMAT = '%Y-%m-%dT%H:%M:%SZ'
  9. DEFAULT_WEBDAV_CONFIG_FILE = "wsgidav.conf"
  10. DEFAULT_TRACIM_CONFIG_FILE = "development.ini"
  11. CONTENT_FRONTEND_URL_SCHEMA = 'workspaces/{workspace_id}/contents/{content_type}/{content_id}' # nopep8
  12. WORKSPACE_FRONTEND_URL_SCHEMA = 'workspaces/{workspace_id}' # nopep8
  13. def get_root_frontend_url(config: 'CFG') -> str:
  14. """
  15. Return website base url with always '/' at the end
  16. """
  17. base_url = ''
  18. if config.WEBSITE_BASE_URL[-1] == '/':
  19. base_url = config.WEBSITE_BASE_URL
  20. else:
  21. base_url = config.WEBSITE_BASE_URL + '/'
  22. return base_url
  23. def get_login_frontend_url(config: 'CFG'):
  24. """
  25. Return login page url
  26. """
  27. return get_root_frontend_url(config) + 'login'
  28. def get_email_logo_frontend_url(config: 'CFG'):
  29. # TODO - G.M - 11-06-2018 - [emailTemplateURL] correct value for email_logo_frontend_url # nopep8
  30. return 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAADAAAAAwCAYAAABXAvmHAAAABmJLR0QA/wD/AP+gvaeTAAAACXBIWXMAAC4jAAAuIwF4pT92AAAAB3RJTUUH4QUTDjMSlsws9AAAB89JREFUaN7tmWtwlFcZx3/PeXeXXLmKQJEGsFZApIUBnaqjUm4VgdYyCUib0cJoKLmHW5EWmlGoQkMICURgEPuhtiYDU7XcyVgKVugULZVCQAwUYyENYAmEXHb3PH7IFjYQNgnZHfnA+bKz+7573v//PPf/C/fWvdWhJWHfMbO4P479OpCAigeowWglPq3ARWfU9AJ7Cb/3MAXZn95dBLKKHsfIQmAUIq7rv6t6gU+BKCAGaEBkN9h88tLe7sgjTdjAJ5Y4OGYMIo80Aw8g4kakJyLxiDiIxABPoMwj5+XP3R0ESpP8WC1H1YuqBj4bUWwIB5gA0d/6/xFQFUpKHBJLHADyU9chTEc1BdEZqJ0O+ofb4xcPyuBIxoBQuHc4ov3w2zrwH0Nd0aBDccwgVPsjEg/4Uc4jegKfvkfMmQ9JSfECkFP0HGJeuu0TrJZhSacg9Xh4Caze2wsXC1B9GugO+BAqUIkC/QIinhYsYoEqYDt+fY2s8WXkFE5CnBIg+vaWtJuQukzy5te2l4Bz2ysTkxchshCRWERMU2BKT4RuiLT0vwagHsENPIyR7zAx2eGybws09AUZEQLHcNS5wMEdB8NjgaKyBNTuQuTLoWOAs6AHUd5DpRzxX0KMBe0M+gA4D4Hdw5mT72LJQORxkM6IfgxSC3wjyHpHEN8M8jKPtYeAq2VgvhHg9A4RvPUIKzCyBVe346SM9LZw1y5KShzOxfUhL60SmMfcNUXgisXYS/iNA/IzhNmBgH4I3E8Cxzpmgby9ffFoMSKTQxA4jZjRpI35qENZbF5Rb1SWgcwM7HsQr28KhZnVd55GPfqTkOCbaB+i2lR1uHa8nHYeda0EygJWGILHGXzndaBgz3CQ5Fb8/hSqr5E7uj4sBXBVSjlqN6FaixKHlS8BxCYtHhs/bUly9JT597U9C30/eRYiU4Nc5Q1gH+AG/gvsxJBH2vg3w9pDPTjqFDGxX0FkKOih2PtHOcbIBiPMMi5XT9eDo9/xlu+rDW2BpX+MAQnOCudRWcWFA3OwncYhrke5VpuK1SrW7+sTVgKv5NYjsr2p6TMY1QEGvhi4mmw89unWXaiX6QQEN1YHuHjpELm5lsxvV2MbhxAdWwCSiNfnCXsb7mc/Ih8hXED8J1T1k6awEAG+R+LCLqEJOL2vInwQOP46VHaSm9QYiI1nMU4xwv2oa3OHs0+LiaHuHKr/wOo/r1yuOQwcDEqVw2KdqITQBFJGelFWojITPz8kY+wmAAp3pWLkJVT7oGwkY3R5REar/Ln1qO7AyIfsLGxQ1eB60E3UDmi9kKWPOwmcvP69cM8kROYCXVB5FW18I4LToeLSbTGnKh2T9MIChLHBOI2lT9srMUDB9s7AM8CApj7H/yaZExsiOd9GnbngcjzOciPyVPOuW0RFB7ZvHnDck4FJgbOpwPoPRHpAdxs782bwQSwei3ty0aC2EVh61ANMutEy63Gi/l0VUfSJC7sIjAnhYYON4x7eNgI9KgeifDPIOyuuDygRWnGuKDcQF6roWrGxbSNgzDAgqBuV6ki7z9WT5y4DR0LdoqoVbSOgZhAi7hv4bXzEFarDG7yqbPmsgN2aovhdLdf+2jqBorIE4JGbIugZCvcsijQHr5ojN1tbVaustUWNNP6S0vy60AR+ut4Ndj7CYzdlgL7AixTtmRnRLOToVNAhQeDfR+yPr+iJrIbf/+pM62l0WP8RKNNCSCCTySuJjgT42MTnv4rqjEDvg6J+VH5T8/qynZSW+ts40EgCQtcQDcsDEN2d9IKe4U2hiR5jmG1ERt4QKrS4pkqK2zmRSQMaUsj6D5cuu3C5ppCY6IQLf2dn8CIjZk4Q+N961VnJvlxf+4Z6v/cojvsDYEQL4BXRHVTXfkysjqPfd3tA6YqOnXx293gnLgvV5xFBVa2iv/b6dEX91p+fbf9ImTnxX6gWAadbILARX+0rbEjxohwCFpBVlH1nyJea2KTFY+OduHUGeaFJoNN3gcwr1smu37qsze16y7pQ4a7pYFYj0iuAvho0jbTxJQBkrB6Ky70VGIDqamA9+WmnWtwrZ20/fN4uQA1rss52Slw8wIOZKg4/QIlTKBfY32DttobSZafDJS0Ka3YXYEx6UCXZid8/n6wJRwHIXrsEI7kB6+xHtQTDn8hLPQuipBb2oJOkgSQBPREuAlupq98cfa6mXsXnOFpfV5uQMBRxlMpP/kJpbmP4tNHC3SNBNiLycJAbleLlOXLGVTB33edRLUBkepAuegyRd1D7Pphh10Wr5mJuMT5fNo67L45NQSUlIBqsYlXqkqaiGy51es3eWQh5CDfmUeufTcaE9QBkFY7AyIuImXxT3bfI7doUvQayFbUDMSZYRHiLq64n2JByOXzvBzLGbkJ1GarnP4OPmBt60Or0v+E381Fdj+ILOhYT4p1ADOhTzcA3EdjeXvCtEwDIGNc0J6vNB/0FDXXbmk9uc05wMSYLq+nAgYArtWJ3keYW0eVUVheE9/3ArbYXkND+mbO2H8gk0AnAlGZAb3WlWpQ/I/o6q9JevWMxIyJdWWZ+V4xnOUaebQH4MVS2I/59eO3bFGbWdEiNiVhrmVncH+NfhMg0IA4RH1bfQnQlq9LKwiYnRbTB/9HmKLrWfg2j94HUoI1/Z3XOOe6te+vuWf8DkM0cb7DOQZgAAAAASUVORK5CYII=' # nopep8'
  31. def get_redis_connection(config: 'CFG') -> Redis:
  32. """
  33. :param config: current app_config
  34. :return: redis connection
  35. """
  36. return Redis(
  37. host=config.EMAIL_SENDER_REDIS_HOST,
  38. port=config.EMAIL_SENDER_REDIS_PORT,
  39. db=config.EMAIL_SENDER_REDIS_DB,
  40. )
  41. def get_rq_queue(redis_connection: Redis, queue_name: str ='default') -> Queue:
  42. """
  43. :param queue_name: name of queue
  44. :return: wanted queue
  45. """
  46. return Queue(name=queue_name, connection=redis_connection)
  47. def cmp_to_key(mycmp):
  48. """
  49. List sort related function
  50. Convert a cmp= function into a key= function
  51. """
  52. class K(object):
  53. def __init__(self, obj, *args):
  54. self.obj = obj
  55. def __lt__(self, other):
  56. return mycmp(self.obj, other.obj) < 0
  57. def __gt__(self, other):
  58. return mycmp(self.obj, other.obj) > 0
  59. def __eq__(self, other):
  60. return mycmp(self.obj, other.obj) == 0
  61. def __le__(self, other):
  62. return mycmp(self.obj, other.obj) <= 0
  63. def __ge__(self, other):
  64. return mycmp(self.obj, other.obj) >= 0
  65. def __ne__(self, other):
  66. return mycmp(self.obj, other.obj) != 0
  67. return K
  68. def current_date_for_filename() -> str:
  69. """
  70. ISO8601 current date, adapted to be used in filename (for
  71. webdav feature for example), with trouble-free characters.
  72. :return: current date as string like "2018-03-19T15.49.27.246592"
  73. """
  74. # INFO - G.M - 19-03-2018 - As ':' is in transform_to_bdd method in
  75. # webdav utils, it may cause trouble. So, it should be replaced to
  76. # a character which will not change in bdd.
  77. return datetime.datetime.now().isoformat().replace(':', '.')
  78. # INFO - G.M - 2018-08-02 - Simple password generator, inspired by
  79. # https://gist.github.com/23maverick23/4131896
  80. ALLOWED_AUTOGEN_PASSWORD_CHAR = string.ascii_letters + \
  81. string.digits + \
  82. string.punctuation
  83. DEFAULT_PASSWORD_GEN_CHAR_LENGTH = 12
  84. def password_generator(
  85. length: int=DEFAULT_PASSWORD_GEN_CHAR_LENGTH,
  86. chars: str=ALLOWED_AUTOGEN_PASSWORD_CHAR
  87. ) -> str:
  88. """
  89. :param length: length of the new password
  90. :param chars: characters allowed
  91. :return: password as string
  92. """
  93. return ''.join(random.choice(chars) for char_number in range(length))