seed.py 842 B

12345678910111213141516171819202122232425262728
  1. from rest_framework_api_key.models import APIKey
  2. from webservice_running.handlers.config import BOTLIST
  3. def create_webservice_seed():
  4. create_api_key_for_handlers()
  5. def create_api_key_for_handlers():
  6. api_key_list = []
  7. for botname in BOTLIST:
  8. keys_to_revoke = APIKey.objects.filter(name="botname", revoked=False)
  9. # Отзываем каждый ключ
  10. for key in keys_to_revoke:
  11. key.revoked = True
  12. key.save()
  13. api_key, key = APIKey.objects.create_key(name=botname)
  14. api_key_list.append([botname, key])
  15. print(f"Initial API Key for {botname} handler created.")
  16. api_key_list = list(map(lambda x: x[0] + '_key=' + x[1], api_key_list))
  17. txt = '\n'.join(api_key_list)
  18. with open('webservice_running/handlers/.env', 'w') as f:
  19. f.write(txt)