def setup_class(cls):
from factories.user import UserFactory
super().setup_class()
session = cls.session_factory()
cls.user_password = hash_password('12345678')
active_user = UserFactory.create(email='{}@example.com'.format(
time.time()),
status='ACTIVE',
password=cls.user_password)
session.refresh(active_user)
cls.user_info = to_dict(active_user)
def write_log(raw_message, level='info', extra=None):
if extra is None:
extra = {}
is_production = True
message = raw_message
try:
extra['message'] = str(message)
if is_production:
getattr(logger, level)(json.dumps(to_dict(extra)))
else:
getattr(logging, level)(message)
except Exception:
logging.error(json.dumps({'message': 'write log failed'}))
def create_admin_am(cls):
password = hash_password('12345678')
session = cls.session_factory
from factories.admin import AdminFactory
am_admin = AdminFactory.create(
email='am_admin_{}@example.com'.format(time.time()),
password=password,
status='ACTIVE',
admin_roles=['AM']
)
session.refresh(am_admin)
cls.am_admin = to_dict(am_admin)
cls.am_admin_token = generate_token({
'admin_id': am_admin.id,
'client_type': 'ADMIN',
'login_at': time.time()
})
def to_json_except_keys(self):
obj = to_dict(self)
for key in self.except_keys():
if key in obj:
del obj[key]
return obj
def to_json(self):
return to_dict(self)