self/configurator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request 'master' (#4) from master into dev

Browse Source 
Reviewed-on: #4
This commit is contained in:
emmatveev 2024-11-24 11:54:32 +03:00
commit 3c851c8e8f
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
app/storage/mongo/__init__.py
View File

@ -18,13 +18,12 @@ def create_indexes():
database.get_collection('configs').create_index([
('stage', 1),
('project', 1),
('name', 1)
])
('name', 1),
], unique=True)
database.get_collection('experiments').create_index([
('stage', 1),
('project', 1),
('name', 1)
])
], unique=True)
database.get_collection('staff').create_index([
('platform_id', 1),
])
], unique=True)

5
app/storage/mongo/staff.py
View File

@ -13,6 +13,7 @@ class Staff(pydantic.BaseModel):
yandex_id: int|None = None
telegram_id: int|None = None
email: str|None = None
telegram_username: str|None = None
async def create(staff: Staff) -> str:
@ -20,8 +21,8 @@ async def create(staff: Staff) -> str:
return result.inserted_id
async def update(platform_id: int, vk_id: int|None, yandex_id: int|None, telegram_id: int|None, email: str|None) -> bool:
result = await collection.update_one({'platform_id': platform_id}, {'$set': {'vk_id': vk_id, 'yandex_id': yandex_id, 'telegram_id': telegram_id, 'email': email}})
async def update(platform_id: int, vk_id: int|None, yandex_id: int|None, telegram_id: int|None, email: str|None, telegram_username: str|None) -> bool:
result = await collection.update_one({'platform_id': platform_id}, {'$set': {'vk_id': vk_id, 'yandex_id': yandex_id, 'telegram_id': telegram_id, 'email': email, 'telegram_username': telegram_username}})
return result.modified_count != 0