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)