24024
af7c11d7f9
- 新增图像生成接口，支持试用、积分和自定义API Key模式 - 实现生成图片结果异步上传至MinIO存储，带重试机制 - 优化积分预扣除和异常退还逻辑，保障用户积分准确 - 添加获取生成历史记录接口，支持时间范围和分页 - 提供本地字典配置接口，支持模型、比例、提示模板和尺寸 - 实现图片批量上传接口，支持S3兼容对象存储 feat(admin): 增加管理员角色管理与权限分配接口 - 实现角色列表查询、角色创建、更新及删除功能 - 增加权限列表查询接口 - 实现用户角色分配接口，便于统一管理用户权限 - 增加系统字典增删查改接口，支持分类过滤和排序 - 权限控制全面覆盖管理接口，保证安全访问 feat(auth): 完善用户登录注册及权限相关接口与页面 - 实现手机号验证码发送及校验功能，保障注册安全 - 支持手机号注册、登录及退出接口，集成日志记录 - 增加修改密码功能，验证原密码后更新 - 提供动态导航菜单接口，基于权限展示不同菜单 - 实现管理界面路由及日志、角色、字典管理页面访问权限控制 - 添加系统日志查询接口，支持关键词和等级筛选 feat(app): 初始化Flask应用并配置蓝图与数据库 - 创建应用程序工厂，加载配置，初始化数据库和Redis客户端 - 注册认证、API及管理员蓝图，整合路由 - 根路由渲染主页模板 - 应用上下文中自动创建数据库表，保证运行环境准备完毕 feat(database): 提供数据库创建与迁移支持脚本 - 新增数据库创建脚本，支持自动检测是否已存在 - 添加数据库表初始化脚本，支持创建和删除所有表 - 实现RBAC权限初始化，包含基础权限和角色创建 - 新增字段手动修复脚本，添加用户API Key和积分字段 - 强制迁移脚本支持清理连接和修复表结构，初始化默认数据及角色分配 feat(config): 新增系统配置参数 - 配置数据库、Redis、Session和MinIO相关参数 - 添加AI接口地址及试用Key配置 - 集成阿里云短信服务配置及开发模式相关参数 feat(extensions): 初始化数据库、Redis和MinIO客户端 - 创建全局SQLAlchemy数据库实例和Redis客户端 - 配置基于boto3的MinIO兼容S3客户端 chore(logs): 添加示例系统日志文件 - 记录用户请求、验证码发送成功与失败的日志信息
313 lines
10 KiB
Python
313 lines
10 KiB
Python
from __future__ import print_function
|
|
|
|
import gc
|
|
import sys
|
|
import unittest
|
|
|
|
from functools import partial
|
|
from unittest import skipUnless
|
|
from unittest import skipIf
|
|
|
|
from greenlet import greenlet
|
|
from greenlet import getcurrent
|
|
from . import TestCase
|
|
from . import PY314
|
|
|
|
try:
|
|
from contextvars import Context
|
|
from contextvars import ContextVar
|
|
from contextvars import copy_context
|
|
# From the documentation:
|
|
#
|
|
# Important: Context Variables should be created at the top module
|
|
# level and never in closures. Context objects hold strong
|
|
# references to context variables which prevents context variables
|
|
# from being properly garbage collected.
|
|
ID_VAR = ContextVar("id", default=None)
|
|
VAR_VAR = ContextVar("var", default=None)
|
|
ContextVar = None
|
|
except ImportError:
|
|
Context = ContextVar = copy_context = None
|
|
|
|
# We don't support testing if greenlet's built-in context var support is disabled.
|
|
@skipUnless(Context is not None, "ContextVar not supported")
|
|
class ContextVarsTests(TestCase):
|
|
def _new_ctx_run(self, *args, **kwargs):
|
|
return copy_context().run(*args, **kwargs)
|
|
|
|
def _increment(self, greenlet_id, callback, counts, expect):
|
|
ctx_var = ID_VAR
|
|
if expect is None:
|
|
self.assertIsNone(ctx_var.get())
|
|
else:
|
|
self.assertEqual(ctx_var.get(), expect)
|
|
ctx_var.set(greenlet_id)
|
|
for _ in range(2):
|
|
counts[ctx_var.get()] += 1
|
|
callback()
|
|
|
|
def _test_context(self, propagate_by):
|
|
# pylint:disable=too-many-branches
|
|
ID_VAR.set(0)
|
|
|
|
callback = getcurrent().switch
|
|
counts = dict((i, 0) for i in range(5))
|
|
|
|
lets = [
|
|
greenlet(partial(
|
|
partial(
|
|
copy_context().run,
|
|
self._increment
|
|
) if propagate_by == "run" else self._increment,
|
|
greenlet_id=i,
|
|
callback=callback,
|
|
counts=counts,
|
|
expect=(
|
|
i - 1 if propagate_by == "share" else
|
|
0 if propagate_by in ("set", "run") else None
|
|
)
|
|
))
|
|
for i in range(1, 5)
|
|
]
|
|
|
|
for let in lets:
|
|
if propagate_by == "set":
|
|
let.gr_context = copy_context()
|
|
elif propagate_by == "share":
|
|
let.gr_context = getcurrent().gr_context
|
|
|
|
for i in range(2):
|
|
counts[ID_VAR.get()] += 1
|
|
for let in lets:
|
|
let.switch()
|
|
|
|
if propagate_by == "run":
|
|
# Must leave each context.run() in reverse order of entry
|
|
for let in reversed(lets):
|
|
let.switch()
|
|
else:
|
|
# No context.run(), so fine to exit in any order.
|
|
for let in lets:
|
|
let.switch()
|
|
|
|
for let in lets:
|
|
self.assertTrue(let.dead)
|
|
# When using run(), we leave the run() as the greenlet dies,
|
|
# and there's no context "underneath". When not using run(),
|
|
# gr_context still reflects the context the greenlet was
|
|
# running in.
|
|
if propagate_by == 'run':
|
|
self.assertIsNone(let.gr_context)
|
|
else:
|
|
self.assertIsNotNone(let.gr_context)
|
|
|
|
|
|
if propagate_by == "share":
|
|
self.assertEqual(counts, {0: 1, 1: 1, 2: 1, 3: 1, 4: 6})
|
|
else:
|
|
self.assertEqual(set(counts.values()), set([2]))
|
|
|
|
def test_context_propagated_by_context_run(self):
|
|
self._new_ctx_run(self._test_context, "run")
|
|
|
|
def test_context_propagated_by_setting_attribute(self):
|
|
self._new_ctx_run(self._test_context, "set")
|
|
|
|
def test_context_not_propagated(self):
|
|
self._new_ctx_run(self._test_context, None)
|
|
|
|
def test_context_shared(self):
|
|
self._new_ctx_run(self._test_context, "share")
|
|
|
|
def test_break_ctxvars(self):
|
|
let1 = greenlet(copy_context().run)
|
|
let2 = greenlet(copy_context().run)
|
|
let1.switch(getcurrent().switch)
|
|
let2.switch(getcurrent().switch)
|
|
# Since let2 entered the current context and let1 exits its own, the
|
|
# interpreter emits:
|
|
# RuntimeError: cannot exit context: thread state references a different context object
|
|
let1.switch()
|
|
|
|
def test_not_broken_if_using_attribute_instead_of_context_run(self):
|
|
let1 = greenlet(getcurrent().switch)
|
|
let2 = greenlet(getcurrent().switch)
|
|
let1.gr_context = copy_context()
|
|
let2.gr_context = copy_context()
|
|
let1.switch()
|
|
let2.switch()
|
|
let1.switch()
|
|
let2.switch()
|
|
|
|
def test_context_assignment_while_running(self):
|
|
# pylint:disable=too-many-statements
|
|
ID_VAR.set(None)
|
|
|
|
def target():
|
|
self.assertIsNone(ID_VAR.get())
|
|
self.assertIsNone(gr.gr_context)
|
|
|
|
# Context is created on first use
|
|
ID_VAR.set(1)
|
|
self.assertIsInstance(gr.gr_context, Context)
|
|
self.assertEqual(ID_VAR.get(), 1)
|
|
self.assertEqual(gr.gr_context[ID_VAR], 1)
|
|
|
|
# Clearing the context makes it get re-created as another
|
|
# empty context when next used
|
|
old_context = gr.gr_context
|
|
gr.gr_context = None # assign None while running
|
|
self.assertIsNone(ID_VAR.get())
|
|
self.assertIsNone(gr.gr_context)
|
|
ID_VAR.set(2)
|
|
self.assertIsInstance(gr.gr_context, Context)
|
|
self.assertEqual(ID_VAR.get(), 2)
|
|
self.assertEqual(gr.gr_context[ID_VAR], 2)
|
|
|
|
new_context = gr.gr_context
|
|
getcurrent().parent.switch((old_context, new_context))
|
|
# parent switches us back to old_context
|
|
|
|
self.assertEqual(ID_VAR.get(), 1)
|
|
gr.gr_context = new_context # assign non-None while running
|
|
self.assertEqual(ID_VAR.get(), 2)
|
|
|
|
getcurrent().parent.switch()
|
|
# parent switches us back to no context
|
|
self.assertIsNone(ID_VAR.get())
|
|
self.assertIsNone(gr.gr_context)
|
|
gr.gr_context = old_context
|
|
self.assertEqual(ID_VAR.get(), 1)
|
|
|
|
getcurrent().parent.switch()
|
|
# parent switches us back to no context
|
|
self.assertIsNone(ID_VAR.get())
|
|
self.assertIsNone(gr.gr_context)
|
|
|
|
gr = greenlet(target)
|
|
|
|
with self.assertRaisesRegex(AttributeError, "can't delete context attribute"):
|
|
del gr.gr_context
|
|
|
|
self.assertIsNone(gr.gr_context)
|
|
old_context, new_context = gr.switch()
|
|
self.assertIs(new_context, gr.gr_context)
|
|
self.assertEqual(old_context[ID_VAR], 1)
|
|
self.assertEqual(new_context[ID_VAR], 2)
|
|
self.assertEqual(new_context.run(ID_VAR.get), 2)
|
|
gr.gr_context = old_context # assign non-None while suspended
|
|
gr.switch()
|
|
self.assertIs(gr.gr_context, new_context)
|
|
gr.gr_context = None # assign None while suspended
|
|
gr.switch()
|
|
self.assertIs(gr.gr_context, old_context)
|
|
gr.gr_context = None
|
|
gr.switch()
|
|
self.assertIsNone(gr.gr_context)
|
|
|
|
# Make sure there are no reference leaks
|
|
gr = None
|
|
gc.collect()
|
|
# Python 3.14 elides reference counting operations
|
|
# in some cases. See https://github.com/python/cpython/pull/130708
|
|
self.assertEqual(sys.getrefcount(old_context), 2 if not PY314 else 1)
|
|
self.assertEqual(sys.getrefcount(new_context), 2 if not PY314 else 1)
|
|
|
|
def test_context_assignment_different_thread(self):
|
|
import threading
|
|
VAR_VAR.set(None)
|
|
ctx = Context()
|
|
|
|
is_running = threading.Event()
|
|
should_suspend = threading.Event()
|
|
did_suspend = threading.Event()
|
|
should_exit = threading.Event()
|
|
holder = []
|
|
|
|
def greenlet_in_thread_fn():
|
|
VAR_VAR.set(1)
|
|
is_running.set()
|
|
should_suspend.wait(10)
|
|
VAR_VAR.set(2)
|
|
getcurrent().parent.switch()
|
|
holder.append(VAR_VAR.get())
|
|
|
|
def thread_fn():
|
|
gr = greenlet(greenlet_in_thread_fn)
|
|
gr.gr_context = ctx
|
|
holder.append(gr)
|
|
gr.switch()
|
|
did_suspend.set()
|
|
should_exit.wait(10)
|
|
gr.switch()
|
|
del gr
|
|
greenlet() # trigger cleanup
|
|
|
|
thread = threading.Thread(target=thread_fn, daemon=True)
|
|
thread.start()
|
|
is_running.wait(10)
|
|
gr = holder[0]
|
|
|
|
# Can't access or modify context if the greenlet is running
|
|
# in a different thread
|
|
with self.assertRaisesRegex(ValueError, "running in a different"):
|
|
getattr(gr, 'gr_context')
|
|
with self.assertRaisesRegex(ValueError, "running in a different"):
|
|
gr.gr_context = None
|
|
|
|
should_suspend.set()
|
|
did_suspend.wait(10)
|
|
|
|
# OK to access and modify context if greenlet is suspended
|
|
self.assertIs(gr.gr_context, ctx)
|
|
self.assertEqual(gr.gr_context[VAR_VAR], 2)
|
|
gr.gr_context = None
|
|
|
|
should_exit.set()
|
|
thread.join(10)
|
|
|
|
self.assertEqual(holder, [gr, None])
|
|
|
|
# Context can still be accessed/modified when greenlet is dead:
|
|
self.assertIsNone(gr.gr_context)
|
|
gr.gr_context = ctx
|
|
self.assertIs(gr.gr_context, ctx)
|
|
|
|
# Otherwise we leak greenlets on some platforms.
|
|
# XXX: Should be able to do this automatically
|
|
del holder[:]
|
|
gr = None
|
|
thread = None
|
|
|
|
def test_context_assignment_wrong_type(self):
|
|
g = greenlet()
|
|
with self.assertRaisesRegex(TypeError,
|
|
"greenlet context must be a contextvars.Context or None"):
|
|
g.gr_context = self
|
|
|
|
|
|
@skipIf(Context is not None, "ContextVar supported")
|
|
class NoContextVarsTests(TestCase):
|
|
def test_contextvars_errors(self):
|
|
let1 = greenlet(getcurrent().switch)
|
|
self.assertFalse(hasattr(let1, 'gr_context'))
|
|
with self.assertRaises(AttributeError):
|
|
getattr(let1, 'gr_context')
|
|
|
|
with self.assertRaises(AttributeError):
|
|
let1.gr_context = None
|
|
|
|
let1.switch()
|
|
|
|
with self.assertRaises(AttributeError):
|
|
getattr(let1, 'gr_context')
|
|
|
|
with self.assertRaises(AttributeError):
|
|
let1.gr_context = None
|
|
|
|
del let1
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|