class UserCache(UserMixin): """用户表缓存""" userid: str username: str alternative_id: str homepage: str permissions: list def get_id(self): """使用替代用户ID,当账号信息发生变化时,替代ID更改为新的随机生成的值,确保他们的旧身份验证会话不再有效。""" return self.alternative_id @staticmethod def add_user_cache(alternative_id: str, **kwargs): """添加user缓存""" user_key = f"{PROJECT_NAME}-{alternative_id}" values = {"alternative_id": alternative_id} values.update(kwargs) redis_client.hmset(user_key, values) redis_client.expire(user_key, SIGNIN_LIFETIME) return True @staticmethod def del_user_cache(alternative_id: str): """删除user缓存""" redis_client.delete(f"{PROJECT_NAME}-{alternative_id}") return True @staticmethod def get_user_cache(alternative_id) -> object: """获取user缓存""" user_cache_dict = redis_client.hgetall(f"{PROJECT_NAME}-{alternative_id}") if user_cache_dict: user_cache = UserCache() user_cache.userid = user_cache_dict.get('userid') user_cache.username = user_cache_dict.get('username') user_cache.alternative_id = user_cache_dict.get('alternative_id') user_cache.homepage = user_cache_dict.get('homepage') user_cache.permissions = user_cache_dict.get('permissions').split(",") # redis里存的是字符串,转为列表 return user_cache 请帮我优化这段代码,并提出建议,python3.8
时间: 2024-02-16 15:05:25 浏览: 129
rsync同步时出现rsync: failed to set times on “xxxx”: Operation not permitted
5星 · 资源好评率100%
以下是我对这段代码的优化和建议:
1. 类的属性最好在 `__init__` 方法中初始化,而不是在类外部初始化。
2. `redis_client.hmset()` 可以使用 `redis_client.hset()` 替代,因为只有一个 key-value。
3. 在 `redis_client.hgetall()` 之前可以先检查一下 key 是否存在,可以避免返回空字典的情况。
4. `redis_client.expire()` 可以设置过期时间的同时返回一个 bool 值表示是否设置成功,可以用来判断是否需要执行下一步操作。
5. `permissions` 属性的默认值应该是一个空列表,而不是 None。
6. 在 `get_user_cache` 方法中,`user_cache_dict.get('permissions')` 可能为 None,应该先检查一下是否为 None,然后再进行 split 操作。
7. 可以将 `redis_client` 对象作为参数传入方法中,而不是在方法内部直接使用全局变量。
8. 可以考虑使用 `typing` 模块来指定函数的参数类型和返回值类型。
优化后的代码如下:
```
from typing import List
import redis
class UserCache(UserMixin):
def __init__(self, userid='', username='', alternative_id='', homepage='', permissions: List[str] = []):
self.userid = userid
self.username = username
self.alternative_id = alternative_id
self.homepage = homepage
self.permissions = permissions
def get_id(self):
return self.alternative_id
@staticmethod
def add_user_cache(alternative_id: str, redis_client: redis.Redis, **kwargs):
user_key = f"{PROJECT_NAME}-{alternative_id}"
values = {"alternative_id": alternative_id}
values.update(kwargs)
redis_client.hset(user_key, mapping=values)
return redis_client.expire(user_key, SIGNIN_LIFETIME)
@staticmethod
def del_user_cache(alternative_id: str, redis_client: redis.Redis):
redis_client.delete(f"{PROJECT_NAME}-{alternative_id}")
return True
@staticmethod
def get_user_cache(alternative_id: str, redis_client: redis.Redis) -> object:
user_key = f"{PROJECT_NAME}-{alternative_id}"
if not redis_client.exists(user_key):
return None
user_cache_dict = redis_client.hgetall(user_key)
user_cache = UserCache()
user_cache.userid = user_cache_dict.get('userid', '')
user_cache.username = user_cache_dict.get('username', '')
user_cache.alternative_id = user_cache_dict.get('alternative_id', '')
user_cache.homepage = user_cache_dict.get('homepage', '')
user_cache.permissions = user_cache_dict.get('permissions', '').split(",") if user_cache_dict.get('permissions') else []
return user_cache
```
建议:
1. 可以考虑将 `redis_client` 对象封装成一个类,方便在多个方法中共用。
2. 可以考虑使用 `@classmethod` 和 `@staticmethod` 来定义类方法和静态方法,增加代码的可读性和可维护性。
3. 可以考虑使用 `logging` 模块来记录日志,方便调试和错误排查。
阅读全文