89 lines
2.5 KiB
Python
89 lines
2.5 KiB
Python
#!/usr/bin/python
|
||
# -*- coding: utf-8 -*-
|
||
# @version : 1.0
|
||
# @Create Time : 2021/10/18 22:18
|
||
# @File : crud.py
|
||
# @IDE : PyCharm
|
||
# @desc : 数据库 增删改查操作
|
||
|
||
import random
|
||
from motor.motor_asyncio import AsyncIOMotorDatabase
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from . import models, schemas
|
||
from core.crud import DalBase
|
||
from core.mongo_manage import MongoManage
|
||
|
||
|
||
class LoginRecordDal(DalBase):
|
||
|
||
def __init__(self, db: AsyncSession):
|
||
super(LoginRecordDal, self).__init__()
|
||
self.db = db
|
||
self.model = models.VadminLoginRecord
|
||
self.schema = schemas.LoginRecordSimpleOut
|
||
|
||
async def get_user_distribute(self) -> list[dict]:
|
||
"""
|
||
获取用户登录分布情况
|
||
高德经纬度查询：https://lbs.amap.com/tools/picker
|
||
|
||
{
|
||
name: '北京',
|
||
center: [116.407394, 39.904211],
|
||
total: 20
|
||
}
|
||
|
||
:return: List[dict]
|
||
"""
|
||
result = [{
|
||
"name": '北京',
|
||
"center": [116.407394, 39.904211],
|
||
},
|
||
{
|
||
"name": '重庆',
|
||
"center": [106.551643, 29.562849],
|
||
},
|
||
{
|
||
"name": '郑州',
|
||
"center": [113.778584, 34.759197],
|
||
},
|
||
{
|
||
"name": '南京',
|
||
"center": [118.796624, 32.059344],
|
||
},
|
||
{
|
||
"name": '武汉',
|
||
"center": [114.304569, 30.593354],
|
||
},
|
||
{
|
||
"name": '乌鲁木齐',
|
||
"center": [87.616824, 43.825377],
|
||
},
|
||
{
|
||
"name": '新乡',
|
||
"center": [113.92679, 35.303589],
|
||
}]
|
||
for data in result:
|
||
assert isinstance(data, dict)
|
||
data["total"] = random.randint(2, 80)
|
||
return result
|
||
|
||
|
||
class SMSSendRecordDal(DalBase):
|
||
|
||
def __init__(self, db: AsyncSession):
|
||
super(SMSSendRecordDal, self).__init__()
|
||
self.db = db
|
||
self.model = models.VadminSMSSendRecord
|
||
self.schema = schemas.SMSSendRecordSimpleOut
|
||
|
||
|
||
class OperationRecordDal(MongoManage):
|
||
|
||
def __init__(self, db: AsyncIOMotorDatabase):
|
||
super(OperationRecordDal, self).__init__()
|
||
self.db = db
|
||
self.collection = db["operation_record"]
|
||
self.schema = schemas.OperationRecordSimpleOut
|
||
self.is_object_id = True
|