init
This commit is contained in:
commit
4f34f5b8be
48
.gitignore
vendored
Normal file
48
.gitignore
vendored
Normal file
@ -0,0 +1,48 @@
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
*.class
|
||||
*.so
|
||||
.Python
|
||||
env/
|
||||
build/
|
||||
develop-eggs/
|
||||
dist/
|
||||
downloads/
|
||||
eggs/
|
||||
.eggs/
|
||||
lib/
|
||||
lib64/
|
||||
parts/
|
||||
sdist/
|
||||
var/
|
||||
*.egg-info/
|
||||
.installed.cfg
|
||||
*.egg
|
||||
*.manifest
|
||||
*.spec
|
||||
pip-log.txt
|
||||
pip-delete-this-directory.txt
|
||||
htmlcov/
|
||||
.tox/
|
||||
.coverage
|
||||
.coverage.*
|
||||
.cache
|
||||
nosetests.xml
|
||||
coverage.xml
|
||||
*.cover
|
||||
.hypothesis/
|
||||
.pytest_cache/
|
||||
*.mo
|
||||
*.pot
|
||||
*.log
|
||||
local_settings.py
|
||||
instance/
|
||||
.webassets-cache
|
||||
.env
|
||||
.venv
|
||||
venv/
|
||||
ENV/
|
||||
.vscode/
|
||||
.idea/
|
||||
*.swp
|
||||
.DS_Store
|
0
Dockerfile
Normal file
0
Dockerfile
Normal file
263
README.md
Normal file
263
README.md
Normal file
@ -0,0 +1,263 @@
|
||||
# ExamService
|
||||
|
||||
题库小程序服务端 - FastAPI实现
|
||||
|
||||
## 项目结构
|
||||
|
||||
```
|
||||
ExamService/
|
||||
├── alembic/ # 数据库迁移相关
|
||||
├── mooc/ # 主应用目录
|
||||
│ ├── api/ # API路由
|
||||
│ │ └── v1/ # API v1版本
|
||||
│ ├── core/ # 核心配置
|
||||
│ ├── crud/ # 数据库操作
|
||||
│ ├── db/ # 数据库
|
||||
│ ├── models/ # 数据库模型
|
||||
│ ├── schemas/ # Pydantic模型
|
||||
│ └── utils/ # 工具函数
|
||||
├── tests/ # 测试目录
|
||||
└── [配置文件]
|
||||
```
|
||||
|
||||
|
||||
## 安装
|
||||
|
||||
### 1. 安装 Miniconda
|
||||
|
||||
在 Ubuntu 上安装 Miniconda,执行以下步骤:
|
||||
|
||||
```bash
|
||||
# 下载 Miniconda 安装脚本
|
||||
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
|
||||
|
||||
# 给安装脚本添加执行权限
|
||||
chmod +x Miniconda3-latest-Linux-x86_64.sh
|
||||
|
||||
# 运行安装脚本
|
||||
./Miniconda3-latest-Linux-x86_64.sh
|
||||
|
||||
# 按照提示完成安装,选择安装目录并接受许可协议
|
||||
|
||||
# 激活 Miniconda
|
||||
source ~/.bashrc
|
||||
# 创建名为 'mooc' 的虚拟环境,Python 版本大于等于 3.13
|
||||
conda create --name mooc python>=3.13
|
||||
# 激活 'mooc' 环境
|
||||
conda activate mooc
|
||||
# 安装依赖
|
||||
pip install -r requirements.txt
|
||||
```
|
||||
|
||||
### 安装mysql开发库
|
||||
|
||||
```bash
|
||||
# 更新包列表
|
||||
apt-get update
|
||||
|
||||
# 安装 MySQL 开发包和其他必要的包
|
||||
apt-get install -y python3-dev default-libmysqlclient-dev build-essential pkg-config
|
||||
|
||||
|
||||
pip install mysqlclient pymysql cryptography
|
||||
```
|
||||
|
||||
## 运行
|
||||
|
||||
```bash
|
||||
uvicorn main:app --reload
|
||||
```
|
||||
|
||||
## 初始化数据库
|
||||
|
||||
```bash
|
||||
alembic init alembic
|
||||
alembic revision --autogenerate -m "Initial migration"
|
||||
alembic upgrade head
|
||||
```
|
||||
|
||||
## 添加数据库表
|
||||
|
||||
### 1.创建模型文件:
|
||||
|
||||
**因为表格太多,可以把同类型表放在同一个文件,文件命名规则:ims_account_wechats、ims_account_wxapp、ims_account_xzapp等表格去除前缀和后缀,即为account.py**
|
||||
|
||||
```python
|
||||
# 这个模型类AccountWechats用于映射ims_account_wechats表:
|
||||
# 1. 在ORM层中代表数据库表结构,方便CRUD操作。
|
||||
# 2. 在FastAPI中引用该模型时,可以使用Pydantic自动转换ORM对象到响应模型。
|
||||
# 3. 创建模型时需确保表名(__tablename__)与真实的数据库表名保持一致,否则查询和插入可能失败。
|
||||
# 4. 字段类型要与数据库中定义的列类型严格匹配,避免出现不兼容或异常。
|
||||
# 5. 通过Primary Key(acid)唯一标识记录,确保ORM能正确追踪和更新该对象。
|
||||
# 6. config.orm_mode = True:允许直接把SQLAlchemy模型实例转换为Pydantic模型对象。
|
||||
|
||||
|
||||
# filepath: ExamService/mooc/models/account.py
|
||||
from sqlalchemy import Column, Integer, String, SmallInteger
|
||||
from mooc.db.database import Base
|
||||
# 类名命名规则: 去除ims前缀后采用 帕斯卡命名法,无连接符每个单词手写大写
|
||||
# AccountWechats模型用于映射数据库表 ims_account_wechats
|
||||
class AccountWechats(Base):
|
||||
__tablename__ = "ims_account_wechats"
|
||||
|
||||
acid = Column(Integer, primary_key=True)
|
||||
uniacid = Column(Integer, nullable=False)
|
||||
token = Column(String(32), nullable=False)
|
||||
encodingaeskey = Column(String(255), nullable=False)
|
||||
level = Column(SmallInteger, nullable=False)
|
||||
name = Column(String(30), nullable=False)
|
||||
account = Column(String(30), nullable=False)
|
||||
original = Column(String(50), nullable=False)
|
||||
signature = Column(String(100), nullable=False)
|
||||
country = Column(String(10), nullable=False)
|
||||
province = Column(String(3), nullable=False)
|
||||
city = Column(String(15), nullable=False)
|
||||
username = Column(String(30), nullable=False)
|
||||
password = Column(String(32), nullable=False)
|
||||
lastupdate = Column(Integer, nullable=False)
|
||||
key = Column(String(50), nullable=False)
|
||||
secret = Column(String(50), nullable=False)
|
||||
styleid = Column(Integer, nullable=False)
|
||||
subscribeurl = Column(String(120), nullable=False)
|
||||
auth_refresh_token = Column(String(255), nullable=False)
|
||||
|
||||
class Config:
|
||||
orm_mode = True # 允许与Pydantic的ORM功能兼容,直接读取数据库模型对象
|
||||
```
|
||||
|
||||
### 2.创建schema:
|
||||
```python
|
||||
# filepath: ExamService/mooc/schemas/account.py
|
||||
|
||||
from pydantic import BaseModel
|
||||
from typing import Optional
|
||||
|
||||
class AccountWechatsBase(BaseModel):
|
||||
"""
|
||||
数据模型基类: AccountWechatsBase
|
||||
用于描述基础字段的类型、用途和注意点。
|
||||
- 注意: 必填字段都在这里声明,每个字段都对应数据库中的一列。
|
||||
"""
|
||||
uniacid: int # 微信公众号/小程序的关联ID
|
||||
token: str # 访问接口时使用的token
|
||||
encodingaeskey: str # 用于消息加解密的AES密钥
|
||||
level: int # 认证等级,如订阅号/服务号
|
||||
name: str # 微信公众号名称
|
||||
account: str # 公众号帐号,如微信号
|
||||
original: str # 原始ID(通常以gh_开头)
|
||||
signature: str # 公众号信息签名
|
||||
country: str # 国家
|
||||
province: str # 省份
|
||||
city: str # 城市
|
||||
username: str # 后台登录用户名
|
||||
password: str # 后台登录密码
|
||||
lastupdate: int # 最后一次更新的时间戳
|
||||
key: str # 开发者Key
|
||||
secret: str # 开发者Secret
|
||||
styleid: int # 模板样式ID
|
||||
subscribeurl: str # 订阅链接
|
||||
auth_refresh_token: str # 用来刷新授权的token
|
||||
|
||||
class AccountWechatsCreate(AccountWechatsBase):
|
||||
"""
|
||||
用于创建新微信帐号记录:
|
||||
- 继承自AccountWechatsBase, 不额外添加字段
|
||||
- 仅表示此Schema专用于'创建'场景
|
||||
"""
|
||||
|
||||
# 用于创建新记录,包含所有必填字段
|
||||
pass
|
||||
|
||||
class AccountWechatsUpdate(BaseModel):
|
||||
"""
|
||||
用于更新已有微信帐号记录:
|
||||
- 只包含可选字段,未在此处的内容将保持不变
|
||||
- 注意: exclude_unset=True 可以避免更新空值
|
||||
"""
|
||||
token: Optional[str] # 可选更新token
|
||||
encodingaeskey: Optional[str]# 可选更新AES key
|
||||
|
||||
class AccountWechats(AccountWechatsBase):
|
||||
"""
|
||||
表示完整的微信帐号记录:
|
||||
- acid: 数据库主键ID
|
||||
- 包含所有字段的最终模型,ORM转换时使用
|
||||
"""
|
||||
acid: int # 表中的主键ID
|
||||
|
||||
class Config:
|
||||
orm_mode = True # 允许与ORM对象进行直接转换
|
||||
```
|
||||
|
||||
### 3.创建 CRUD:
|
||||
```python
|
||||
# filepath: ExamService/mooc/crud/crud_account.py
|
||||
from sqlalchemy.orm import Session
|
||||
from typing import Optional
|
||||
|
||||
from mooc.models.account import AccountWechats
|
||||
from mooc.schemas.account import AccountWechatsCreate, AccountWechatsUpdate
|
||||
|
||||
class CRUDAccountWechats:
|
||||
"""
|
||||
创建CRUD
|
||||
负责对AccountWechats模型进行增删改查操作的逻辑封装。
|
||||
"""
|
||||
|
||||
def create(self, db: Session, obj_in: AccountWechatsCreate) -> AccountWechats:
|
||||
"""
|
||||
创建记录:将Pydantic的输入数据转换为数据库模型实例并保存。
|
||||
"""
|
||||
db_obj = AccountWechats(**obj_in.dict()) # 将表单数据解包到模型实例
|
||||
db.add(db_obj) # 添加到会话
|
||||
db.commit() # 提交事务
|
||||
db.refresh(db_obj) # 刷新实例,获取数据库中的最新状态
|
||||
return db_obj
|
||||
|
||||
def get(self, db: Session, acid: int) -> Optional[AccountWechats]:
|
||||
"""
|
||||
根据主键 acid 查询单条数据。
|
||||
"""
|
||||
return db.query(AccountWechats).filter(AccountWechats.acid == acid).first()
|
||||
|
||||
def update(
|
||||
self, db: Session, *, db_obj: AccountWechats, obj_in: AccountWechatsUpdate
|
||||
) -> AccountWechats:
|
||||
"""
|
||||
更新记录:只修改传递进来的字段,未设置的字段不动。
|
||||
"""
|
||||
for field, value in obj_in.dict(exclude_unset=True).items():
|
||||
setattr(db_obj, field, value) # 更新模型实例属性
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def delete(self, db: Session, acid: int) -> None:
|
||||
"""
|
||||
删除记录:物理删除或逻辑删除都可在此处实现。
|
||||
"""
|
||||
obj = db.query(AccountWechats).filter(AccountWechats.acid == acid).first()
|
||||
if obj:
|
||||
db.delete(obj)
|
||||
db.commit()
|
||||
|
||||
# 实例化CRUD对象,方便在业务代码中直接引用
|
||||
account_wechats = CRUDAccountWechats()
|
||||
```
|
||||
### 4.注册模型:
|
||||
```python
|
||||
# filepath: ExamService/mooc/db/base.py
|
||||
from mooc.db.database import Base
|
||||
from mooc.models.admin import Admin, Account, AccountWebapp
|
||||
from mooc.models.account import AccountWechats # 新增,将创建好的模型类导入即可
|
||||
```
|
||||
|
||||
### 5.运行代码更新数据库信息
|
||||
```bash
|
||||
python main.py
|
||||
```
|
||||
|
||||
## API文档
|
||||
|
||||
启动服务后访问: http://localhost:2333/docs
|
117
alembic.ini
Normal file
117
alembic.ini
Normal file
@ -0,0 +1,117 @@
|
||||
# A generic, single database configuration.
|
||||
|
||||
[alembic]
|
||||
# path to migration scripts
|
||||
# Use forward slashes (/) also on windows to provide an os agnostic path
|
||||
script_location = alembic
|
||||
|
||||
# template used to generate migration file names; The default value is %%(rev)s_%%(slug)s
|
||||
# Uncomment the line below if you want the files to be prepended with date and time
|
||||
# see https://alembic.sqlalchemy.org/en/latest/tutorial.html#editing-the-ini-file
|
||||
# for all available tokens
|
||||
# file_template = %%(year)d_%%(month).2d_%%(day).2d_%%(hour).2d%%(minute).2d-%%(rev)s_%%(slug)s
|
||||
|
||||
# sys.path path, will be prepended to sys.path if present.
|
||||
# defaults to the current working directory.
|
||||
prepend_sys_path = .
|
||||
|
||||
# timezone to use when rendering the date within the migration file
|
||||
# as well as the filename.
|
||||
# If specified, requires the python>=3.9 or backports.zoneinfo library.
|
||||
# Any required deps can installed by adding `alembic[tz]` to the pip requirements
|
||||
# string value is passed to ZoneInfo()
|
||||
# leave blank for localtime
|
||||
# timezone =
|
||||
|
||||
# max length of characters to apply to the "slug" field
|
||||
# truncate_slug_length = 40
|
||||
|
||||
# set to 'true' to run the environment during
|
||||
# the 'revision' command, regardless of autogenerate
|
||||
# revision_environment = false
|
||||
|
||||
# set to 'true' to allow .pyc and .pyo files without
|
||||
# a source .py file to be detected as revisions in the
|
||||
# versions/ directory
|
||||
# sourceless = false
|
||||
|
||||
# version location specification; This defaults
|
||||
# to alembic/versions. When using multiple version
|
||||
# directories, initial revisions must be specified with --version-path.
|
||||
# The path separator used here should be the separator specified by "version_path_separator" below.
|
||||
# version_locations = %(here)s/bar:%(here)s/bat:alembic/versions
|
||||
|
||||
# version path separator; As mentioned above, this is the character used to split
|
||||
# version_locations. The default within new alembic.ini files is "os", which uses os.pathsep.
|
||||
# If this key is omitted entirely, it falls back to the legacy behavior of splitting on spaces and/or commas.
|
||||
# Valid values for version_path_separator are:
|
||||
#
|
||||
# version_path_separator = :
|
||||
# version_path_separator = ;
|
||||
# version_path_separator = space
|
||||
# version_path_separator = newline
|
||||
version_path_separator = os # Use os.pathsep. Default configuration used for new projects.
|
||||
|
||||
# set to 'true' to search source files recursively
|
||||
# in each "version_locations" directory
|
||||
# new in Alembic version 1.10
|
||||
# recursive_version_locations = false
|
||||
|
||||
# the output encoding used when revision files
|
||||
# are written from script.py.mako
|
||||
# output_encoding = utf-8
|
||||
|
||||
sqlalchemy.url = mysql+pymysql://mooc:zXpPHKhE7A5x6X3A@localhost/mooc
|
||||
|
||||
|
||||
[post_write_hooks]
|
||||
# post_write_hooks defines scripts or Python functions that are run
|
||||
# on newly generated revision scripts. See the documentation for further
|
||||
# detail and examples
|
||||
|
||||
# format using "black" - use the console_scripts runner, against the "black" entrypoint
|
||||
# hooks = black
|
||||
# black.type = console_scripts
|
||||
# black.entrypoint = black
|
||||
# black.options = -l 79 REVISION_SCRIPT_FILENAME
|
||||
|
||||
# lint with attempts to fix using "ruff" - use the exec runner, execute a binary
|
||||
# hooks = ruff
|
||||
# ruff.type = exec
|
||||
# ruff.executable = %(here)s/.venv/bin/ruff
|
||||
# ruff.options = --fix REVISION_SCRIPT_FILENAME
|
||||
|
||||
# Logging configuration
|
||||
[loggers]
|
||||
keys = root,sqlalchemy,alembic
|
||||
|
||||
[handlers]
|
||||
keys = console
|
||||
|
||||
[formatters]
|
||||
keys = generic
|
||||
|
||||
[logger_root]
|
||||
level = WARNING
|
||||
handlers = console
|
||||
qualname =
|
||||
|
||||
[logger_sqlalchemy]
|
||||
level = WARNING
|
||||
handlers =
|
||||
qualname = sqlalchemy.engine
|
||||
|
||||
[logger_alembic]
|
||||
level = INFO
|
||||
handlers =
|
||||
qualname = alembic
|
||||
|
||||
[handler_console]
|
||||
class = StreamHandler
|
||||
args = (sys.stderr,)
|
||||
level = NOTSET
|
||||
formatter = generic
|
||||
|
||||
[formatter_generic]
|
||||
format = %(levelname)-5.5s [%(name)s] %(message)s
|
||||
datefmt = %H:%M:%S
|
1
alembic/README
Normal file
1
alembic/README
Normal file
@ -0,0 +1 @@
|
||||
Generic single-database configuration.
|
78
alembic/env.py
Normal file
78
alembic/env.py
Normal file
@ -0,0 +1,78 @@
|
||||
from logging.config import fileConfig
|
||||
|
||||
from sqlalchemy import engine_from_config
|
||||
from sqlalchemy import pool
|
||||
|
||||
from alembic import context
|
||||
|
||||
# this is the Alembic Config object, which provides
|
||||
# access to the values within the .ini file in use.
|
||||
config = context.config
|
||||
|
||||
# Interpret the config file for Python logging.
|
||||
# This line sets up loggers basically.
|
||||
if config.config_file_name is not None:
|
||||
fileConfig(config.config_file_name)
|
||||
|
||||
# add your model's MetaData object here
|
||||
# for 'autogenerate' support
|
||||
# from myapp import mymodel
|
||||
# target_metadata = mymodel.Base.metadata
|
||||
target_metadata = None
|
||||
|
||||
# other values from the config, defined by the needs of env.py,
|
||||
# can be acquired:
|
||||
# my_important_option = config.get_main_option("my_important_option")
|
||||
# ... etc.
|
||||
|
||||
|
||||
def run_migrations_offline() -> None:
|
||||
"""Run migrations in 'offline' mode.
|
||||
|
||||
This configures the context with just a URL
|
||||
and not an Engine, though an Engine is acceptable
|
||||
here as well. By skipping the Engine creation
|
||||
we don't even need a DBAPI to be available.
|
||||
|
||||
Calls to context.execute() here emit the given string to the
|
||||
script output.
|
||||
|
||||
"""
|
||||
url = config.get_main_option("sqlalchemy.url")
|
||||
context.configure(
|
||||
url=url,
|
||||
target_metadata=target_metadata,
|
||||
literal_binds=True,
|
||||
dialect_opts={"paramstyle": "named"},
|
||||
)
|
||||
|
||||
with context.begin_transaction():
|
||||
context.run_migrations()
|
||||
|
||||
|
||||
def run_migrations_online() -> None:
|
||||
"""Run migrations in 'online' mode.
|
||||
|
||||
In this scenario we need to create an Engine
|
||||
and associate a connection with the context.
|
||||
|
||||
"""
|
||||
connectable = engine_from_config(
|
||||
config.get_section(config.config_ini_section, {}),
|
||||
prefix="sqlalchemy.",
|
||||
poolclass=pool.NullPool,
|
||||
)
|
||||
|
||||
with connectable.connect() as connection:
|
||||
context.configure(
|
||||
connection=connection, target_metadata=target_metadata
|
||||
)
|
||||
|
||||
with context.begin_transaction():
|
||||
context.run_migrations()
|
||||
|
||||
|
||||
if context.is_offline_mode():
|
||||
run_migrations_offline()
|
||||
else:
|
||||
run_migrations_online()
|
26
alembic/script.py.mako
Normal file
26
alembic/script.py.mako
Normal file
@ -0,0 +1,26 @@
|
||||
"""${message}
|
||||
|
||||
Revision ID: ${up_revision}
|
||||
Revises: ${down_revision | comma,n}
|
||||
Create Date: ${create_date}
|
||||
|
||||
"""
|
||||
from typing import Sequence, Union
|
||||
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
${imports if imports else ""}
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision: str = ${repr(up_revision)}
|
||||
down_revision: Union[str, None] = ${repr(down_revision)}
|
||||
branch_labels: Union[str, Sequence[str], None] = ${repr(branch_labels)}
|
||||
depends_on: Union[str, Sequence[str], None] = ${repr(depends_on)}
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
${upgrades if upgrades else "pass"}
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
${downgrades if downgrades else "pass"}
|
0
docker-compose.yml
Normal file
0
docker-compose.yml
Normal file
32
main.py
Executable file
32
main.py
Executable file
@ -0,0 +1,32 @@
|
||||
from fastapi import FastAPI
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
import uvicorn
|
||||
from mooc.api.v1.api import api_router
|
||||
from mooc.core.config import settings
|
||||
from mooc.db.database import init_db
|
||||
|
||||
app = FastAPI(
|
||||
title="ExamService",
|
||||
description="题库小程序服务端API",
|
||||
version="1.0.0"
|
||||
)
|
||||
|
||||
# CORS设置
|
||||
app.add_middleware(
|
||||
CORSMiddleware,
|
||||
allow_origins=["*"],
|
||||
allow_credentials=True,
|
||||
allow_methods=["*"],
|
||||
allow_headers=["*"],
|
||||
)
|
||||
# 初始化数据库
|
||||
init_db()
|
||||
|
||||
app.include_router(api_router, prefix=settings.API_V1_STR)
|
||||
|
||||
@app.get("/")
|
||||
async def root():
|
||||
return {"message": "Welcome to ExamService API"}
|
||||
|
||||
if __name__ == '__main__':
|
||||
uvicorn.run('main:app', host='0.0.0.0', port=2333, reload=True, workers=1)
|
0
mooc/__init__.py
Normal file
0
mooc/__init__.py
Normal file
0
mooc/api/__init__.py
Normal file
0
mooc/api/__init__.py
Normal file
62
mooc/api/deps.py
Normal file
62
mooc/api/deps.py
Normal file
@ -0,0 +1,62 @@
|
||||
from fastapi import Depends, HTTPException, status
|
||||
from sqlalchemy.orm import Session
|
||||
from mooc.db.session import get_db
|
||||
from fastapi.security import OAuth2PasswordBearer
|
||||
import jwt
|
||||
from mooc.core.config import settings
|
||||
from mooc.crud.crud_admin import admin
|
||||
from mooc.models.admin import Admin
|
||||
|
||||
|
||||
|
||||
|
||||
oauth2_scheme = OAuth2PasswordBearer(tokenUrl=f"{settings.API_V1_STR}/login/access-token")
|
||||
|
||||
# 获取当前用户的依赖
|
||||
def get_current_user(
|
||||
db: Session = Depends(get_db),
|
||||
token: str = Depends(oauth2_scheme)
|
||||
) -> Admin:
|
||||
"""
|
||||
验证当前用户的依赖项
|
||||
"""
|
||||
try:
|
||||
# PyJWT的解码方式
|
||||
payload = jwt.decode(
|
||||
token,
|
||||
settings.SECRET_KEY,
|
||||
algorithms=[settings.ALGORITHM]
|
||||
)
|
||||
username: str = payload.get("sub")
|
||||
if not username:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
||||
detail="Could not validate credentials",
|
||||
)
|
||||
except jwt.InvalidTokenError: # PyJWT的异常处理
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
||||
detail="Could not validate credentials",
|
||||
)
|
||||
|
||||
user = admin.get_by_username(db, username=username)
|
||||
if not user:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_404_NOT_FOUND,
|
||||
detail="User not found",
|
||||
)
|
||||
return user
|
||||
|
||||
# 验证超级管理员的依赖
|
||||
def get_current_superuser(
|
||||
current_user: Admin = Depends(get_current_user),
|
||||
) -> Admin:
|
||||
"""
|
||||
验证超级管理员的依赖项
|
||||
"""
|
||||
if current_user.is_delete != 0:
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_403_FORBIDDEN,
|
||||
detail="The user doesn't have enough privileges",
|
||||
)
|
||||
return current_user
|
0
mooc/api/v1/__init__.py
Normal file
0
mooc/api/v1/__init__.py
Normal file
14
mooc/api/v1/api.py
Normal file
14
mooc/api/v1/api.py
Normal file
@ -0,0 +1,14 @@
|
||||
from fastapi import APIRouter
|
||||
from mooc.api.v1.endpoints import admin, wechat
|
||||
|
||||
api_router = APIRouter()
|
||||
|
||||
# עԱ·ɣǰΪ/admins
|
||||
api_router.include_router(
|
||||
admin.admin_router,
|
||||
prefix="/admins",
|
||||
tags=["admins"]
|
||||
)
|
||||
|
||||
# Add WeChat endpoints
|
||||
api_router.include_router(wechat.router, prefix="/wechat", tags=["wechat"])
|
0
mooc/api/v1/endpoints/__init__.py
Normal file
0
mooc/api/v1/endpoints/__init__.py
Normal file
93
mooc/api/v1/endpoints/admin.py
Normal file
93
mooc/api/v1/endpoints/admin.py
Normal file
@ -0,0 +1,93 @@
|
||||
from typing import List
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
from sqlalchemy.orm import Session
|
||||
from mooc.api import deps
|
||||
from mooc.crud.crud_admin import admin
|
||||
from mooc.schemas.admin import Admin, AdminCreate, AdminUpdate
|
||||
|
||||
|
||||
admin_router = APIRouter()
|
||||
@admin_router.get("/", response_model=List[Admin])
|
||||
def read_admins(
|
||||
db: Session = Depends(deps.get_db),
|
||||
# 添加当前用户验证依赖
|
||||
current_user: Admin = Depends(deps.get_current_user),
|
||||
skip: int = 0,
|
||||
limit: int = 100,):
|
||||
"""
|
||||
获取管理员列表
|
||||
需要登录权限
|
||||
"""
|
||||
admins = admin.get_multi(db, skip=skip, limit=limit)
|
||||
return admins
|
||||
|
||||
@admin_router.post("/", response_model=Admin)
|
||||
def create_admin(
|
||||
*,
|
||||
db: Session = Depends(deps.get_db),
|
||||
# 添加超级管理员验证依赖
|
||||
# current_user: Admin = Depends(deps.get_current_superuser),
|
||||
admin_in: AdminCreate,
|
||||
):
|
||||
"""
|
||||
创建新管理员
|
||||
需要超级管理员权限
|
||||
"""
|
||||
admin_obj = admin.get_by_username(db, username=admin_in.username)
|
||||
if admin_obj:
|
||||
raise HTTPException(
|
||||
status_code=400,
|
||||
detail="Username already registered"
|
||||
)
|
||||
return admin.create(db, obj_in=admin_in)
|
||||
|
||||
@admin_router.put("/{id}", response_model=Admin)
|
||||
def update_admin(
|
||||
*,
|
||||
db: Session = Depends(deps.get_db),
|
||||
# 添加超级管理员验证依赖
|
||||
current_user: Admin = Depends(deps.get_current_superuser),
|
||||
id: int,
|
||||
admin_in: AdminUpdate,
|
||||
):
|
||||
"""
|
||||
更新管理员信息
|
||||
需要超级管理员权限
|
||||
"""
|
||||
admin_obj = admin.get(db, id=id)
|
||||
if not admin_obj:
|
||||
raise HTTPException(
|
||||
status_code=404,
|
||||
detail="Admin not found"
|
||||
)
|
||||
return admin.update(db, db_obj=admin_obj, obj_in=admin_in)
|
||||
|
||||
@admin_router.delete("/{id}", response_model=Admin)
|
||||
def delete_admin(
|
||||
*,
|
||||
db: Session = Depends(deps.get_db),
|
||||
# 添加超级管理员验证依赖
|
||||
current_user: Admin = Depends(deps.get_current_superuser),
|
||||
id: int,
|
||||
):
|
||||
"""
|
||||
删除管理员
|
||||
需要超级管理员权限
|
||||
"""
|
||||
admin_obj = admin.get(db, id=id)
|
||||
if not admin_obj:
|
||||
raise HTTPException(
|
||||
status_code=404,
|
||||
detail="Admin not found"
|
||||
)
|
||||
return admin.delete(db, id=id)
|
||||
|
||||
# 添加获取当前用户信息的接口
|
||||
@admin_router.get("/me", response_model=Admin)
|
||||
def read_user_me(
|
||||
current_user: Admin = Depends(deps.get_current_user),
|
||||
):
|
||||
"""
|
||||
获取当前登录用户信息
|
||||
"""
|
||||
return current_user
|
0
mooc/api/v1/endpoints/auth.py
Normal file
0
mooc/api/v1/endpoints/auth.py
Normal file
42
mooc/api/v1/endpoints/wechat.py
Normal file
42
mooc/api/v1/endpoints/wechat.py
Normal file
@ -0,0 +1,42 @@
|
||||
from fastapi import APIRouter, Depends, HTTPException
|
||||
from typing import Dict, Any
|
||||
from mooc.utils.wechat_client import WeChatClient
|
||||
|
||||
router = APIRouter()
|
||||
wechat_client = WeChatClient()
|
||||
|
||||
@router.get("/access_token")
|
||||
async def get_access_token() -> Dict[str, Any]:
|
||||
"""获取微信access token"""
|
||||
try:
|
||||
access_token = await wechat_client.get_access_token()
|
||||
return {"access_token": access_token}
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
|
||||
@router.post("/code2session")
|
||||
async def code_to_session(code: str) -> Dict[str, Any]:
|
||||
"""小程序登录"""
|
||||
try:
|
||||
result = await wechat_client.code2session(code)
|
||||
return result
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
|
||||
@router.post("/qrcode")
|
||||
async def generate_qrcode(scene: str) -> bytes:
|
||||
"""生成小程序码"""
|
||||
try:
|
||||
qr_code = await wechat_client.get_unlimited_qrcode(scene)
|
||||
return qr_code
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
|
||||
@router.post("/send_template")
|
||||
async def send_template_message(data: Dict[str, Any]) -> Dict[str, Any]:
|
||||
"""发送订阅消息"""
|
||||
try:
|
||||
result = await wechat_client.send_template_message(data)
|
||||
return result
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
0
mooc/core/__init__.py
Normal file
0
mooc/core/__init__.py
Normal file
36
mooc/core/config.py
Normal file
36
mooc/core/config.py
Normal file
@ -0,0 +1,36 @@
|
||||
from pydantic_settings import BaseSettings
|
||||
from typing import Optional
|
||||
|
||||
class Settings(BaseSettings):
|
||||
PROJECT_NAME: str = "ExamService"
|
||||
VERSION: str = "1.0.0"
|
||||
API_V1_STR: str = "/api/v1"
|
||||
|
||||
|
||||
MYSQL_USER: str = "mooc"
|
||||
MYSQL_PASSWORD: str = "zXpPHKhE7A5x6X3A"
|
||||
MYSQL_HOST: str = "localhost"
|
||||
MYSQL_PORT: str = "3306"
|
||||
MYSQL_DATABASE: str = "mooc"
|
||||
|
||||
SQLALCHEMY_DATABASE_URI: Optional[str] = None
|
||||
SQLALCHEMY_ECHO: bool = True
|
||||
|
||||
SECRET_KEY: str = "your-secret-key-here" # <20><><EFBFBD><EFBFBD>ʹ<EFBFBD>ø<EFBFBD><C3B8>ӵ<EFBFBD><D3B5><EFBFBD><EFBFBD><EFBFBD>ַ<EFBFBD><D6B7><EFBFBD>
|
||||
ALGORITHM: str = "HS256"
|
||||
ACCESS_TOKEN_EXPIRE_MINUTES: int = 30
|
||||
|
||||
WECHAT_APPID: str = ""
|
||||
WECHAT_APPSECRET: str = ""
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
super().__init__(**kwargs)
|
||||
self.SQLALCHEMY_DATABASE_URI = (
|
||||
f"mysql+pymysql://{self.MYSQL_USER}:{self.MYSQL_PASSWORD}"
|
||||
f"@{self.MYSQL_HOST}:{self.MYSQL_PORT}/{self.MYSQL_DATABASE}"
|
||||
)
|
||||
|
||||
class Config:
|
||||
env_file = ".env"
|
||||
|
||||
settings = Settings()
|
0
mooc/core/events.py
Normal file
0
mooc/core/events.py
Normal file
25
mooc/core/security.py
Normal file
25
mooc/core/security.py
Normal file
@ -0,0 +1,25 @@
|
||||
from datetime import datetime, timedelta
|
||||
from typing import Optional
|
||||
import jwt
|
||||
from mooc.core.config import settings
|
||||
|
||||
def create_access_token(
|
||||
subject: str, expires_delta: Optional[timedelta] = None
|
||||
) -> str:
|
||||
"""
|
||||
´´½¨·ÃÎÊÁîÅÆ
|
||||
"""
|
||||
if expires_delta:
|
||||
expire = datetime.now(datetime.timezone.utc) + expires_delta
|
||||
else:
|
||||
expire = datetime.now(datetime.timezone.utc) + timedelta(
|
||||
minutes=settings.ACCESS_TOKEN_EXPIRE_MINUTES
|
||||
)
|
||||
|
||||
to_encode = {"exp": expire, "sub": str(subject)}
|
||||
encoded_jwt = jwt.encode(
|
||||
to_encode,
|
||||
settings.SECRET_KEY,
|
||||
algorithm=settings.ALGORITHM
|
||||
)
|
||||
return encoded_jwt
|
0
mooc/crud/__init__.py
Normal file
0
mooc/crud/__init__.py
Normal file
34
mooc/crud/crud_account.py
Normal file
34
mooc/crud/crud_account.py
Normal file
@ -0,0 +1,34 @@
|
||||
from sqlalchemy.orm import Session
|
||||
from typing import Optional
|
||||
|
||||
from mooc.models.account import AccountWechats
|
||||
from mooc.schemas.account import AccountWechatsCreate, AccountWechatsUpdate
|
||||
|
||||
class CRUDAccountWechats:
|
||||
def create(self, db: Session, obj_in: AccountWechatsCreate) -> AccountWechats:
|
||||
db_obj = AccountWechats(**obj_in.dict())
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def get(self, db: Session, acid: int) -> Optional[AccountWechats]:
|
||||
return db.query(AccountWechats).filter(AccountWechats.acid == acid).first()
|
||||
|
||||
def update(
|
||||
self, db: Session, *, db_obj: AccountWechats, obj_in: AccountWechatsUpdate
|
||||
) -> AccountWechats:
|
||||
for field, value in obj_in.dict(exclude_unset=True).items():
|
||||
setattr(db_obj, field, value)
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def delete(self, db: Session, acid: int) -> None:
|
||||
obj = db.query(AccountWechats).filter(AccountWechats.acid == acid).first()
|
||||
if obj:
|
||||
db.delete(obj)
|
||||
db.commit()
|
||||
|
||||
account_wechats = CRUDAccountWechats()
|
98
mooc/crud/crud_admin.py
Normal file
98
mooc/crud/crud_admin.py
Normal file
@ -0,0 +1,98 @@
|
||||
from typing import List, Optional
|
||||
from sqlalchemy.orm import Session
|
||||
from mooc.models.admin import Admin,Account,AccountWebapp
|
||||
from mooc.schemas.admin import AdminCreate, AdminUpdate,AccountCreate, AccountUpdate, AccountWebappCreate, AccountWebappUpdate
|
||||
|
||||
|
||||
class CRUDAdmin:
|
||||
def get(self, db: Session, id: int) -> Optional[Admin]:
|
||||
return db.query(Admin).filter(Admin.id == id).first()
|
||||
|
||||
def get_by_username(self, db: Session, username: str) -> Optional[Admin]:
|
||||
return db.query(Admin).filter(Admin.username == username).first()
|
||||
|
||||
def get_multi(
|
||||
self, db: Session, *, skip: int = 0, limit: int = 100
|
||||
) -> List[Admin]:
|
||||
return db.query(Admin).offset(skip).limit(limit).all()
|
||||
|
||||
def create(self, db: Session, *, obj_in: AdminCreate) -> Admin:
|
||||
db_obj = Admin(
|
||||
weid=obj_in.weid,
|
||||
username=obj_in.username,
|
||||
password=obj_in.password, # ע<>⣺ʵ<E2A3BA><CAB5>ʹ<EFBFBD><CAB9>ʱӦ<CAB1>ö<EFBFBD><C3B6><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>й<EFBFBD>ϣ<EFBFBD><CFA3><EFBFBD><EFBFBD>?
|
||||
pcate_id=obj_in.pcate_id,
|
||||
cate_id=obj_in.cate_id,
|
||||
relation_id=obj_in.relation_id,
|
||||
is_delete=obj_in.is_delete
|
||||
)
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def update(
|
||||
self, db: Session, *, db_obj: Admin, obj_in: AdminUpdate
|
||||
) -> Admin:
|
||||
update_data = obj_in.dict(exclude_unset=True)
|
||||
for field, value in update_data.items():
|
||||
setattr(db_obj, field, value)
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def delete(self, db: Session, *, id: int) -> Admin:
|
||||
obj = db.query(Admin).get(id)
|
||||
if obj:
|
||||
obj.is_delete = 1
|
||||
db.add(obj)
|
||||
db.commit()
|
||||
return obj
|
||||
|
||||
admin = CRUDAdmin()
|
||||
|
||||
|
||||
|
||||
class CRUDAccount:
|
||||
def get_by_uniacid(self, db: Session, *, uniacid: int) -> Optional[Account]:
|
||||
return db.query(Account).filter(Account.uniacid == uniacid).first()
|
||||
|
||||
account = CRUDAccount()
|
||||
|
||||
|
||||
|
||||
|
||||
class CRUDAccountWebapp:
|
||||
def create(self, db: Session, *, obj_in: AccountWebappCreate) -> AccountWebapp:
|
||||
db_obj = AccountWebapp(**obj_in.dict())
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def get(self, db: Session, acid: int) -> Optional[AccountWebapp]:
|
||||
return db.query(AccountWebapp).filter(AccountWebapp.acid == acid).first()
|
||||
|
||||
def update(
|
||||
self,
|
||||
db: Session,
|
||||
*,
|
||||
db_obj: AccountWebapp,
|
||||
obj_in: AccountWebappUpdate
|
||||
) -> AccountWebapp:
|
||||
data = obj_in.dict(exclude_unset=True)
|
||||
for field, value in data.items():
|
||||
setattr(db_obj, field, value)
|
||||
db.add(db_obj)
|
||||
db.commit()
|
||||
db.refresh(db_obj)
|
||||
return db_obj
|
||||
|
||||
def delete(self, db: Session, acid: int) -> None:
|
||||
obj = db.query(AccountWebapp).filter(AccountWebapp.acid == acid).first()
|
||||
if obj:
|
||||
db.delete(obj)
|
||||
db.commit()
|
||||
|
||||
account_webapp = CRUDAccountWebapp()
|
0
mooc/db/__init__.py
Normal file
0
mooc/db/__init__.py
Normal file
5
mooc/db/base.py
Normal file
5
mooc/db/base.py
Normal file
@ -0,0 +1,5 @@
|
||||
from mooc.db.database import Base
|
||||
|
||||
# 导入所有模型以注册到Base.metadata
|
||||
from mooc.models.admin import Admin, Account, AccountWebapp
|
||||
from mooc.models.account import AccountWechats
|
29
mooc/db/database.py
Normal file
29
mooc/db/database.py
Normal file
@ -0,0 +1,29 @@
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.ext.declarative import declarative_base
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
from mooc.core.config import settings
|
||||
|
||||
# 创建数据库引擎
|
||||
engine = create_engine(
|
||||
settings.SQLALCHEMY_DATABASE_URI,
|
||||
pool_pre_ping=True,
|
||||
echo=settings.SQLALCHEMY_ECHO
|
||||
)
|
||||
|
||||
# 创建会话工厂
|
||||
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
||||
|
||||
# 创建基类
|
||||
Base = declarative_base()
|
||||
|
||||
def init_db():
|
||||
"""初始化数据库"""
|
||||
Base.metadata.create_all(bind=engine)
|
||||
|
||||
def get_db():
|
||||
"""获取数据库会话"""
|
||||
db = SessionLocal()
|
||||
try:
|
||||
yield db
|
||||
finally:
|
||||
db.close()
|
29
mooc/db/session.py
Normal file
29
mooc/db/session.py
Normal file
@ -0,0 +1,29 @@
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
from mooc.core.config import settings
|
||||
from typing import Generator
|
||||
from mooc.db.base import Base
|
||||
|
||||
# 创建数据库引擎
|
||||
engine = create_engine(
|
||||
settings.SQLALCHEMY_DATABASE_URI,
|
||||
pool_pre_ping=True,
|
||||
echo=settings.SQLALCHEMY_ECHO
|
||||
)
|
||||
|
||||
# 创建表
|
||||
def init_db():
|
||||
Base.metadata.create_all(bind=engine)
|
||||
|
||||
# 创建会话工厂
|
||||
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
||||
|
||||
def get_db() -> Generator:
|
||||
"""
|
||||
获取数据库会话的依赖项
|
||||
"""
|
||||
try:
|
||||
db = SessionLocal()
|
||||
yield db
|
||||
finally:
|
||||
db.close()
|
0
mooc/models/__init__.py
Normal file
0
mooc/models/__init__.py
Normal file
29
mooc/models/account.py
Normal file
29
mooc/models/account.py
Normal file
@ -0,0 +1,29 @@
|
||||
from sqlalchemy import Column, Integer, String, SmallInteger
|
||||
from mooc.db.database import Base
|
||||
|
||||
class AccountWechats(Base):
|
||||
__tablename__ = "ims_account_wechats"
|
||||
|
||||
acid = Column(Integer, primary_key=True)
|
||||
uniacid = Column(Integer, nullable=False)
|
||||
token = Column(String(32), nullable=False)
|
||||
encodingaeskey = Column(String(255), nullable=False)
|
||||
level = Column(SmallInteger, nullable=False)
|
||||
name = Column(String(30), nullable=False)
|
||||
account = Column(String(30), nullable=False)
|
||||
original = Column(String(50), nullable=False)
|
||||
signature = Column(String(100), nullable=False)
|
||||
country = Column(String(10), nullable=False)
|
||||
province = Column(String(3), nullable=False)
|
||||
city = Column(String(15), nullable=False)
|
||||
username = Column(String(30), nullable=False)
|
||||
password = Column(String(32), nullable=False)
|
||||
lastupdate = Column(Integer, nullable=False)
|
||||
key = Column(String(50), nullable=False)
|
||||
secret = Column(String(50), nullable=False)
|
||||
styleid = Column(Integer, nullable=False)
|
||||
subscribeurl = Column(String(120), nullable=False)
|
||||
auth_refresh_token = Column(String(255), nullable=False)
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
51
mooc/models/admin.py
Normal file
51
mooc/models/admin.py
Normal file
@ -0,0 +1,51 @@
|
||||
from sqlalchemy import Column, Integer, String, Text, SmallInteger
|
||||
from sqlalchemy.sql import func
|
||||
|
||||
import time
|
||||
from mooc.db.database import Base
|
||||
|
||||
|
||||
class Admin(Base):
|
||||
__tablename__ = "ims_goouc_fullexam_admin"
|
||||
|
||||
id = Column(Integer, primary_key=True, autoincrement=True)
|
||||
weid = Column(String(150), nullable=False)
|
||||
username = Column(String(255), nullable=False)
|
||||
password = Column(String(255), nullable=False)
|
||||
createtime = Column(Integer, nullable=False, default=lambda: int(time.time()))
|
||||
logintime = Column(Integer, nullable=False, default=lambda: int(time.time()))
|
||||
is_delete = Column(SmallInteger, nullable=False, default=1, comment='0正常 1禁用')
|
||||
pcate_id = Column(Integer, nullable=False, comment='一级类目id')
|
||||
cate_id = Column(Integer, nullable=False, comment='二级级类目id')
|
||||
relation_id = Column(Text, nullable=False, comment='关联试卷ID')
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
class Account(Base):
|
||||
__tablename__ = "ims_account"
|
||||
|
||||
acid = Column(Integer, primary_key=True, autoincrement=True)
|
||||
uniacid = Column(Integer, nullable=False, index=True)
|
||||
hash = Column(String(8), nullable=False)
|
||||
type = Column(SmallInteger, nullable=False)
|
||||
isconnect = Column(SmallInteger, nullable=False)
|
||||
isdeleted = Column(SmallInteger, nullable=False)
|
||||
endtime = Column(Integer, nullable=False)
|
||||
send_account_expire_status = Column(SmallInteger, nullable=False)
|
||||
send_api_expire_status = Column(SmallInteger, nullable=False)
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
|
||||
|
||||
class AccountWebapp(Base):
|
||||
__tablename__ = "ims_account_webapp"
|
||||
|
||||
acid = Column(Integer, primary_key=True)
|
||||
uniacid = Column(Integer, index=True)
|
||||
name = Column(String(255))
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
0
mooc/schemas/__init__.py
Normal file
0
mooc/schemas/__init__.py
Normal file
37
mooc/schemas/account.py
Normal file
37
mooc/schemas/account.py
Normal file
@ -0,0 +1,37 @@
|
||||
from pydantic import BaseModel
|
||||
from typing import Optional
|
||||
|
||||
class AccountWechatsBase(BaseModel):
|
||||
uniacid: int
|
||||
token: str
|
||||
encodingaeskey: str
|
||||
level: int
|
||||
name: str
|
||||
account: str
|
||||
original: str
|
||||
signature: str
|
||||
country: str
|
||||
province: str
|
||||
city: str
|
||||
username: str
|
||||
password: str
|
||||
lastupdate: int
|
||||
key: str
|
||||
secret: str
|
||||
styleid: int
|
||||
subscribeurl: str
|
||||
auth_refresh_token: str
|
||||
|
||||
class AccountWechatsCreate(AccountWechatsBase):
|
||||
pass
|
||||
|
||||
class AccountWechatsUpdate(BaseModel):
|
||||
token: Optional[str]
|
||||
encodingaeskey: Optional[str]
|
||||
|
||||
|
||||
class AccountWechats(AccountWechatsBase):
|
||||
acid: int
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
78
mooc/schemas/admin.py
Normal file
78
mooc/schemas/admin.py
Normal file
@ -0,0 +1,78 @@
|
||||
from typing import Optional
|
||||
from pydantic import BaseModel
|
||||
from datetime import datetime
|
||||
|
||||
# 基础Admin模型
|
||||
class AdminBase(BaseModel):
|
||||
weid: str
|
||||
username: str
|
||||
pcate_id: int
|
||||
cate_id: int
|
||||
relation_id: str
|
||||
is_delete: int = 1
|
||||
|
||||
# 创建Admin时的请求模型
|
||||
class AdminCreate(AdminBase):
|
||||
password: str
|
||||
|
||||
# 更新Admin时的请求模型
|
||||
class AdminUpdate(BaseModel):
|
||||
weid: Optional[str] = None
|
||||
username: Optional[str] = None
|
||||
password: Optional[str] = None
|
||||
pcate_id: Optional[int] = None
|
||||
cate_id: Optional[int] = None
|
||||
relation_id: Optional[str] = None
|
||||
is_delete: Optional[int] = None
|
||||
|
||||
# Admin响应模型
|
||||
class Admin(AdminBase):
|
||||
id: int
|
||||
createtime: int
|
||||
logintime: int
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
|
||||
class AccountBase(BaseModel):
|
||||
uniacid: int
|
||||
hash: str
|
||||
type: int
|
||||
isconnect: int
|
||||
isdeleted: int
|
||||
endtime: int
|
||||
send_account_expire_status: int
|
||||
send_api_expire_status: int
|
||||
|
||||
class AccountCreate(AccountBase):
|
||||
pass
|
||||
|
||||
class AccountUpdate(AccountBase):
|
||||
pass
|
||||
|
||||
class Account(AccountBase):
|
||||
acid: int
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
class AccountWebappBase(BaseModel):
|
||||
uniacid: Optional[int]
|
||||
name: Optional[str]
|
||||
|
||||
class AccountWebappCreate(AccountWebappBase):
|
||||
pass
|
||||
|
||||
class AccountWebappUpdate(AccountWebappBase):
|
||||
pass
|
||||
|
||||
class AccountWebapp(AccountWebappBase):
|
||||
acid: int
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
0
mooc/utils/__init__.py
Normal file
0
mooc/utils/__init__.py
Normal file
54
mooc/utils/wechat_client.py
Normal file
54
mooc/utils/wechat_client.py
Normal file
@ -0,0 +1,54 @@
|
||||
import json
|
||||
import httpx
|
||||
from typing import Optional, Dict, Any
|
||||
from mooc.core.config import settings
|
||||
|
||||
class WeChatClient:
|
||||
def __init__(self, appid: str = None, appsecret: str = None):
|
||||
self.appid = appid or settings.WECHAT_APPID
|
||||
self.appsecret = appsecret or settings.WECHAT_APPSECRET
|
||||
self._access_token = ""
|
||||
|
||||
async def _get(self, url: str) -> Dict[str, Any]:
|
||||
async with httpx.AsyncClient() as client:
|
||||
response = await client.get(url, verify=False)
|
||||
return response.json()
|
||||
|
||||
async def _post(self, url: str, data: Dict[str, Any]) -> Dict[str, Any]:
|
||||
async with httpx.AsyncClient() as client:
|
||||
response = await client.post(url, json=data, verify=False)
|
||||
return response.json()
|
||||
|
||||
async def get_access_token(self) -> str:
|
||||
"""获取access token"""
|
||||
url = f"https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid={self.appid}&secret={self.appsecret}"
|
||||
result = await self._get(url)
|
||||
if "access_token" in result:
|
||||
self._access_token = result["access_token"]
|
||||
return self._access_token
|
||||
raise Exception(f"Failed to get access token: {result}")
|
||||
|
||||
async def code2session(self, code: str) -> Dict[str, Any]:
|
||||
"""小程序登录凭证校验"""
|
||||
url = f"https://api.weixin.qq.com/sns/jscode2session?appid={self.appid}&secret={self.appsecret}&js_code={code}&grant_type=authorization_code"
|
||||
result = await self._get(url)
|
||||
if "errcode" in result and result["errcode"] != 0:
|
||||
raise Exception(f"Code2Session failed: {result['errmsg']}")
|
||||
return result
|
||||
|
||||
async def get_unlimited_qrcode(self, scene: str, access_token: str = None) -> bytes:
|
||||
"""获取小程序码"""
|
||||
if not access_token:
|
||||
access_token = await self.get_access_token()
|
||||
url = f"https://api.weixin.qq.com/wxa/getwxacodeunlimit?access_token={access_token}"
|
||||
data = {"scene": scene}
|
||||
async with httpx.AsyncClient() as client:
|
||||
response = await client.post(url, json=data, verify=False)
|
||||
return response.content
|
||||
|
||||
async def send_template_message(self, data: Dict[str, Any], access_token: str = None) -> Dict[str, Any]:
|
||||
"""发送订阅消息"""
|
||||
if not access_token:
|
||||
access_token = await self.get_access_token()
|
||||
url = f"https://api.weixin.qq.com/cgi-bin/message/wxopen/template/send?access_token={access_token}"
|
||||
return await self._post(url, data)
|
20
requirements.txt
Normal file
20
requirements.txt
Normal file
@ -0,0 +1,20 @@
|
||||
passlib[bcrypt]
|
||||
alembic>=1.7.1
|
||||
httpx>=0.19.0
|
||||
redis>=4.0.2
|
||||
python-dotenv>=0.19.0
|
||||
sqlalchemy>=2.0.36
|
||||
fastapi>=0.115.6
|
||||
pyjwt>=2.10.1
|
||||
requests>=2.32.3
|
||||
pydantic>=2.10.4
|
||||
passlib>=1.7.4
|
||||
uvicorn>=0.34.0
|
||||
pytest>=7.4.4
|
||||
Pillow>=11.0.0
|
||||
numpy>=2.2.1
|
||||
python-multipart>=0.0.20
|
||||
pymysql>=1.1.1
|
||||
cryptography>=42.0.5
|
||||
mysqlclient>=2.2.6
|
||||
pydantic-settings>=2.7.0
|
0
tests/__init__.py
Normal file
0
tests/__init__.py
Normal file
0
tests/conftest.py
Normal file
0
tests/conftest.py
Normal file
Loading…
Reference in New Issue
Block a user