def create_user(user: User, is_admin: bool = False) -> User:
"""
创建用户
:param user: User 用户ORM
:param is_admin: bool 是否admin
创建头像信息,创建用户基本信息
"""
user.roles = Role.get_by_user_default(is_admin)
avator = Storages(**ADMIN_AVATOR) if is_admin else Storages(**USER_AVATOR)
user.userinfo.avator = avator
return user.save()
def init_development_users(password: Optional[str] = None) -> None:
"""
初始化数据
"""
su_role = Role.get_by_name(name="SuperUser")
if not password:
print("请输入初始超级用户密码")
password = getpass("Password:"******"wisdom",
password=password,
email="*****@*****.**",
phonenum="1234567",
active=True,
confirmed_at=datetime.utcnow(),
)
avator = Storages(
name="AdminAvator.jpg",
storetype="avator",
saved=True,
filetype="image/jpeg",
path="default/AdminAvator.jpg",
uid=1,
store=load_avator_from_path("default/AdminAvator.jpg"),
)
UserInfo.create(user=root, avator=avator)
root.roles.append(su_role)
root.save()
def init():
"""
初始化数据
"""
su_role = Role.get_by_name(name="SuperUser")
# create super user
root = User.create(
username="******",
password="******",
email="*****@*****.**",
phonenum="1234567",
active=True,
confirmed_at=datetime.utcnow(),
)
avator = Storages(
name="AdminAvator.jpg",
storetype="avator",
saved=True,
filetype="image/jpeg",
path="default/AdminAvator.jpg",
uid=1,
)
UserInfo.create(user=root, avator=avator)
root.roles.append(su_role)
root.save()
def add_storage(store: FileStorage) -> "Storages":
# pylint: disable=W0621
from smorest_sfs.services.storages.handlers import StorageFactory
from smorest_sfs.modules.storages.models import Storages
factory = StorageFactory(
Storages(name="test.txt", storetype="foo",
store=store)) # type: ignore
storage: "Storages" = factory.save()
return storage
def test_model_load(self) -> None:
from smorest_sfs.services.storages.handlers import StorageFactory
from smorest_sfs.modules.storages.models import Storages
factory = StorageFactory(self.storage)
factory.save()
storage = Storages.find_or_fail(self.storage.id_)
for _ in range(3):
storage.as_stream()
assert storage.store.read() == b"abc"
def create_user(user: User, is_admin: bool = False) -> User:
"""
创建用户
:param user: User 用户ORM
:param is_admin: bool 是否admin
创建头像信息,创建用户基本信息
"""
user.roles = Role.get_by_user_default(is_admin)
avator = (
Storages( # type: ignore
store=load_avator_from_path(ADMIN_AVATOR["path"]),
saved=True,
**ADMIN_AVATOR) if is_admin else Storages( # type: ignore
store=load_avator_from_path(USER_AVATOR["path"]),
saved=True,
**USER_AVATOR))
user.userinfo.update(avator=avator)
return user
def test_model_update(self) -> None:
from smorest_sfs.services.storages.handlers import StorageFactory
from smorest_sfs.modules.storages.models import Storages
factory = StorageFactory(self.storage)
factory.save()
factory.update(name="t.txt", store=self.next_store)
storage = Storages.find_or_fail(self.storage.id_)
storage.store = None
assert storage.store.read(
) == b"efg" and storage.name == "t.txt" # type: ignore
def storage(store: FileStorage) -> "Storages":
# pylint: disable=W0621
from smorest_sfs.modules.storages.models import Storages
return Storages(name="test.txt", storetype="foo",
store=store) # type: ignore
def add_storage(store):
factory = StorageFactory(
Storages(name="test.txt", storetype="foo", store=store))
return factory.save()
def storage(store):
return Storages(name="test.txt", storetype="foo", store=store)