392dba39ea
Necesitaba un ratelimit que estuviera disponible en todos los workers, cosa que aparentemente la libreria que estaba usando no tomaba en cuenta asi que la mejor idea que tuve es aprovechar el cache y usarlo para generar el ratelimit :3 asi que como todos los workers se comunican con el mismo cache, todos van a compartir el mismo lock
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
"""Modulo de funciones utilitarias"""
|
|
import json
|
|
import logging
|
|
import functools
|
|
|
|
from multiprocessing.dummy import Pool
|
|
|
|
from django.http import JsonResponse
|
|
|
|
from pygments import highlight
|
|
from pygments.lexers import JsonLexer # pylint: disable=no-name-in-module
|
|
from pygments.formatters import TerminalTrueColorFormatter # pylint: disable=no-name-in-module
|
|
from utils.ratelimit import ratelimit
|
|
|
|
_log = logging.getLogger('utils')
|
|
_log.addHandler(logging.NullHandler())
|
|
|
|
|
|
def pretty_print_json(json_input):
|
|
"""Formats and prints json to stdout with colors using pygments"""
|
|
formatted_json = json.dumps(json_input, indent=2)
|
|
print(highlight(formatted_json, JsonLexer(), TerminalTrueColorFormatter()))
|
|
|
|
|
|
def message_response(status, error_message):
|
|
"""Sends an error response with the status code of the error and an explanation"""
|
|
json_message = {
|
|
'status_code': status,
|
|
'message': error_message
|
|
}
|
|
|
|
return JsonResponse(json_message, status=status)
|
|
|
|
|
|
def require_json(function):
|
|
"""Decorator to make a view only accept a json body"""
|
|
@functools.wraps(function)
|
|
def decorator(*args, **kwargs):
|
|
try:
|
|
received_json = json.loads(args[0].body)
|
|
return function(*args, **kwargs, received_json=received_json)
|
|
except json.JSONDecodeError:
|
|
_log.warning('Function %s got a non json request body', function.__name__)
|
|
return message_response(400, 'Se envío json no valido')
|
|
|
|
return decorator
|
|
|
|
|
|
def expected_keys(keys, dictionary):
|
|
"""Verifica que un diccionario contiene todas las keys de una lista"""
|
|
for key in keys:
|
|
if key not in dictionary:
|
|
return f'No se encuentra {key}'
|
|
return None
|
|
|
|
|
|
def replace_key(dictionary, old, new):
|
|
"""Remplaza una key en un diccionario
|
|
|
|
En esencia crea una nueva key con los contenidos de la antigua y elimina la antigua
|
|
|
|
Esto lo hace inplace por lo que el diccionario es modificado
|
|
|
|
TODO que cree un nuevo diccionario, es mas bonito el codigo asi
|
|
"""
|
|
dictionary[new] = dictionary[old]
|
|
del dictionary[old]
|
|
|
|
|
|
def sanitize_keys(dictionary):
|
|
"""Remplaza las keys de un diccionario que contienen '-' ya que el diccionario pretende
|
|
ser json y acceder a json con esos caracteres en javascript es incomodo
|
|
|
|
El diccionario es modificado inplace pero de todas formas se retorna
|
|
"""
|
|
for key in list(dictionary.keys()):
|
|
if '-' in key and key in dictionary:
|
|
new_key = key.replace('-', '_')
|
|
replace_key(dictionary, key, new_key)
|
|
return dictionary
|