python-gRPC

1
import asyncio
2
import os
3
import sys
4
from datetime import datetime as dt
5
from sqlalchemy import text
6

7
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
8
sys.path.append(os.path.dirname(SCRIPT_DIR))
9

10
from src.db.db_connector import async_session_maker  # noqa
11

12

13
async def load_test_users() -> None:
14
    async with async_session_maker() as session:
15
        query = text(
16
            """
17
            INSERT INTO public.user (
18
                fullname,
19
                password,
20
                email,
21
                is_admin,
22
                is_moderator,
23
                is_active,
24
                date_at
25
            )
26
            VALUES (
27
                :fullname,
28
                :password,
29
                :email,
30
                :is_admin,
31
                :is_moderator,
32
                :is_active,
33
                :date_at
34
            )
35
        """
36
        )
37
        values = [
38
            {
39
                "fullname": "adminov",
40
                "email": "admin@admin.admin",
41
                "password": "pass",
42
                "is_admin": True,
43
                "is_moderator": True,
44
                "is_active": True,
45
                "date_at": dt.now(),
46
            },
47
            {
48
                "fullname": "adminov_1",
49
                "email": "admin_1@admin.admin",
50
                "password": "pass_1",
51
                "is_admin": True,
52
                "is_moderator": True,
53
                "is_active": True,
54
                "date_at": dt.now(),
55
            },
56
        ]
57
        print("Run script. Loads Users")
58
        for value in values:
59
            await session.execute(query, value)
60
            await session.commit()
61

62
    print("Test user loaded!")
63

64

65
if __name__ == "__main__":
66
    asyncio.run(load_test_users())
67

68

69
"""
70
"""
71