73 lines
2.7 KiB
Python
73 lines
2.7 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
# @version : 1.0
|
|
# @Creaet Time : 2022/7/7 13:41
|
|
# @File : login.py
|
|
# @IDE : PyCharm
|
|
# @desc : 登录记录模型
|
|
import json
|
|
|
|
from application.settings import LOGIN_LOG_RECORD
|
|
from apps.vadmin.auth.utils.validation import LoginForm
|
|
from utils.ip_manage import IPManage
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from db.db_base import BaseModel
|
|
from sqlalchemy import Column, String, Boolean, TEXT
|
|
from fastapi import Request
|
|
from user_agents import parse
|
|
|
|
|
|
class VadminLoginRecord(BaseModel):
|
|
__tablename__ = "vadmin_record_login"
|
|
__table_args__ = ({'comment': '登录记录表'})
|
|
|
|
telephone = Column(String(50), index=True, nullable=False, comment="手机号")
|
|
status = Column(Boolean, default=True, comment="是否登录成功")
|
|
platform = Column(String(8), comment="登陆平台")
|
|
login_method = Column(String(8), comment="认证方式")
|
|
ip = Column(String(50), comment="登陆地址")
|
|
address = Column(String(255), comment="登陆地点")
|
|
country = Column(String(255), comment="国家")
|
|
province = Column(String(255), comment="县")
|
|
city = Column(String(255), comment="城市")
|
|
county = Column(String(255), comment="区/县")
|
|
operator = Column(String(255), comment="运营商")
|
|
postal_code = Column(String(255), comment="邮政编码")
|
|
area_code = Column(String(255), comment="地区区号")
|
|
browser = Column(String(50), comment="浏览器")
|
|
system = Column(String(50), comment="操作系统")
|
|
response = Column(TEXT, comment="响应信息")
|
|
request = Column(TEXT, comment="请求信息")
|
|
|
|
@classmethod
|
|
async def create_login_record(cls, db: AsyncSession, data: LoginForm, status: bool, req: Request, resp: dict):
|
|
"""
|
|
创建登录记录
|
|
@return:
|
|
"""
|
|
if not LOGIN_LOG_RECORD:
|
|
return None
|
|
header = {}
|
|
for k, v in req.headers.items():
|
|
header[k] = v
|
|
body = json.loads((await req.body()).decode())
|
|
user_agent = parse(req.headers.get("user-agent"))
|
|
system = f"{user_agent.os.family} {user_agent.os.version_string}"
|
|
browser = f"{user_agent.browser.family} {user_agent.browser.version_string}"
|
|
ip = IPManage(req.client.host)
|
|
location = await ip.parse()
|
|
params = json.dumps({"body": body, "headers": header})
|
|
obj = VadminLoginRecord(
|
|
**location.dict(),
|
|
telephone=data.telephone,
|
|
status=status,
|
|
browser=browser,
|
|
system=system,
|
|
response=json.dumps(resp),
|
|
request=params,
|
|
platform=data.platform,
|
|
login_method=data.method
|
|
)
|
|
db.add(obj)
|
|
await db.flush()
|