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