InteCat/integrase.py at master · MetroReviews/InteCat · 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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
# Packages
import metro_integrase as metro
import os
from uuid import uuid4
from fastapi import FastAPI, Request
from fastapi.encoders import jsonable_encoder
from fastapi.responses import ORJSONResponse
from sqlalchemy import Table, MetaData, create_engine, Column, String, JSON, Boolean, update, select, insert
from dotenv import load_dotenv
load_dotenv()

# Configure Fast API
app = FastAPI()

# Configure Metro Reviews
MetroReviews = metro.Metro(domain="https://metro.select-list.xyz", list_id=os.getenv("LIST_ID"), secret_key=os.getenv("SECRET_KEY"), app=app)

# Database
engine = create_engine(os.getenv("DATABASE_URI"))
metadata = MetaData()
connection = engine.connect()

# Bots schema
bots = Table(
    "bots",
    metadata,
    Column("bot_id", String),
    Column("uuid", String),
    Column("username", String),
    Column("description", String),
    Column("long_description", String),
    Column("state", String),
    Column("flags", JSON),
    Column("owner", String),
    Column("extra_owners", JSON),
    Column("library", String),
    Column("nsfw", Boolean),
    Column("tags", JSON),
    Column("invite", String),
)

# Claim
@MetroReviews.claim()
async def claim(request: Request, bot: metro.Bot):
    data = select([bots]).where(
        (bots.columns.bot_id == bot.bot_id)
    )

    botExists = False

    for row in connection.execute(data):
        botExists = True

    if botExists == True:
        connection.execute(update(bots).where(bots.columns.bot_id == bot.bot_id).values(state="CLAIMED"))
        res = {
            "content": "Bot Claimed",
            "done": True
        }
    else:
        uuid = uuid4()
        state = "CLAIMED"
        connection.execute(insert(bots).values(bot_id=bot.bot_id, uuid=uuid, username=bot.username, description=bot.description, long_description=bot.long_description, state=state, flags=["METRO"], owner=bot.owner, extra_owners=bot.extra_owners, library=bot.library, nsfw=bot.nsfw, tags=bot.tags, invite=bot.invite))
        res = {
            "content": "Bot Added and Claimed",
            "done": True
        }

    return ORJSONResponse(content=jsonable_encoder(res))

# Unclaim
@MetroReviews.unclaim()
async def unclaim(request: Request, bot: metro.Bot):
    data = select([bots]).where(
        (bots.columns.bot_id == bot.bot_id)
    )

    botExists = False

    for row in connection.execute(data):
        botExists = True

    if botExists == True:
        connection.execute(update(bots).where(bots.columns.bot_id == bot.bot_id).values(state="AWAITING_REVIEW"))
        res = {
            "content": "Bot Unclaimed",
            "done": True
        }
    else:
        uuid = uuid4()
        state = "AWAITING_REVIEW"
        connection.execute(insert(bots).values(bot_id=bot.bot_id, uuid=uuid, username=bot.username, description=bot.description, long_description=bot.long_description, state=state, flags=["METRO"], owner=bot.owner, extra_owners=bot.extra_owners, library=bot.library, nsfw=bot.nsfw, tags=bot.tags, invite=bot.invite))
        res = {
            "content": "Bot Added and Unclaimed",
            "done": True
        }

    return ORJSONResponse(content=jsonable_encoder(res))

# Approve
@MetroReviews.approve()
async def approve(request: Request, bot: metro.Bot):
    data = select([bots]).where(
        (bots.columns.bot_id == bot.bot_id)
    )

    botExists = False

    for row in connection.execute(data):
        botExists = True

    if botExists == True:
        connection.execute(update(bots).where(bots.columns.bot_id == bot.bot_id).values(state="APPROVED"))
        res = {
            "content": "Bot Approved",
            "done": True
        }
    else:
        uuid = uuid4()
        state = "APPROVED"
        connection.execute(insert(bots).values(bot_id=bot.bot_id, uuid=uuid, username=bot.username, description=bot.description, long_description=bot.long_description, state=state, flags=["METRO"], owner=bot.owner, extra_owners=bot.extra_owners, library=bot.library, nsfw=bot.nsfw, tags=bot.tags, invite=bot.invite))
        res = {
            "content": "Bot Added and Approved",
            "done": True
        }

    return ORJSONResponse(content=jsonable_encoder(res))


# Deny
@MetroReviews.deny()
async def deny(request: Request, bot: metro.Bot):
    data = select([bots]).where(
        (bots.columns.bot_id == bot.bot_id)
    )

    botExists = False

    for row in connection.execute(data):
        botExists = True

    if botExists == True:
        connection.execute(update(bots).where(bots.columns.bot_id == bot.bot_id).values(state="DENIED"))
        res = {
            "content": "Bot Denied",
            "done": True
        }
    else:
        uuid = uuid4()
        state = "DENIED"
        connection.execute(insert(bots).values(bot_id=bot.bot_id, uuid=uuid, username=bot.username, description=bot.description, long_description=bot.long_description, state=state, flags=["METRO"], owner=bot.owner, extra_owners=bot.extra_owners, library=bot.library, nsfw=bot.nsfw, tags=bot.tags, invite=bot.invite))
        res = {
            "content": "Bot Added and Denied",
            "done": True
        }

    return ORJSONResponse(content=jsonable_encoder(res))

# Startup Event
@app.on_event("startup")
async def startup():
    print("Started!")
    await MetroReviews.register_api_urls()