ochabot/main.py at main · sofyan48/ochabot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
import os, uvicorn, argparse
from pkg import utils
from dotenv import load_dotenv
from config.logger import logging_config
from alembic import command
from alembic.config import Config
from pkg.logger.logging import logger

try:
    result = load_dotenv()
except Exception as e:
    logger.error("Cannot load .env file", {
        "eror": e
    })


def migrate():
    parser = argparse.ArgumentParser()
    subparsers = parser.add_subparsers(dest="subcommand", help="All Command")
    migrate_up_parser = subparsers.add_parser("up", help="Running migrate")
    migrate_create_parser = subparsers.add_parser("create", help="Create migrate")
    migrate_down_parser = subparsers.add_parser("down", help="Downgrade migrate")
    migrate_create_parser.add_argument("migration_name", help="Nama untuk migrasi baru")
    migrate_down_parser.add_argument("migration_revision", help="Migrate revision")
    migrate_up_parser.add_argument("migration_revision", help="Migrate revision")
    args = parser.parse_args()
    alembic_cfg = Config("alembic.ini")

    print(args.subcommand)

    if args.subcommand == "up":
        migrate_revision = args.migration_revision
        command.upgrade(alembic_cfg, migrate_revision)
        exit(0)

    if args.subcommand == "create":
        migration_name = args.migration_name
        command.revision(alembic_cfg, message=migration_name, autogenerate=False)  # Membuat migrasi baru
        exit(0)

    if args.subcommand == "down":
        migrate_revision = args.migration_revision
        command.downgrade(alembic_cfg, migrate_revision)
        exit(0)


def http_development_mode():
    parser = argparse.ArgumentParser()
    parser.add_argument("command", help="Perintah yang ingin Anda jalankan")
    args = parser.parse_args()
    if args.command == "serve":
        host = os.getenv("APP_HOST", "0.0.0.0")
        port = int(os.getenv("APP_PORT", "8080"))

        is_reload = False
        if utils.environment_transform() == 'loc':
            is_reload = True

        logger.info("Starting Server", {
            "host": host,
            "port": port,
        })
        uvicorn.run(
            app="app:app",
            host=host,
            port=port,
            reload=is_reload,
            access_log=True,
            reload_excludes=["mydb/*", "minio/*", "chroma/*"]
        )
        exit(0)

def http():
    parser = argparse.ArgumentParser()
    parser.add_argument("command", help="Perintah yang ingin Anda jalankan")
    args = parser.parse_args()
    if args.command == "serve":
        host = os.getenv("APP_HOST", "0.0.0.0")
        port = int(os.getenv("APP_PORT", "8080"))

        is_reload = False
        if utils.environment_transform() == 'loc':
            is_reload = True

        logger.info("Starting Server", {
            "host": host,
            "port": port,
        })
        uvicorn.run(
            app="app:app",
            host=host,
            port=port,
            reload=is_reload,
            log_config=logging_config,
            access_log=True,
            reload_excludes=["mydb/*", "minio/*", "chroma/*"]
        )
        exit(0)

def main():
    http()