randydev commited on
Commit
d0bbe79
1 Parent(s): 3373ef5

Delete database.py

Browse files
Files changed (1) hide show
  1. database.py +0 -117
database.py DELETED
@@ -1,117 +0,0 @@
1
- #!/usr/bin/env python
2
- # -*- coding: utf-8 -*-
3
- # Copyright 2020-2023 (c) Randy W @xtdevs, @xtsea
4
- #
5
- # from : https://github.com/TeamKillerX
6
- # Channel : @RendyProjects
7
- # This program is free software: you can redistribute it and/or modify
8
- # it under the terms of the GNU Affero General Public License as published by
9
- # the Free Software Foundation, either version 3 of the License, or
10
- # (at your option) any later version.
11
- #
12
- # This program is distributed in the hope that it will be useful,
13
- # but WITHOUT ANY WARRANTY; without even the implied warranty of
14
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15
- # GNU Affero General Public License for more details.
16
- #
17
- # You should have received a copy of the GNU Affero General Public License
18
- # along with this program. If not, see <https://www.gnu.org/licenses/>.
19
-
20
- import os
21
- from dotenv import load_dotenv
22
- from pymongo import MongoClient
23
-
24
- load_dotenv()
25
- MONGO_URL = os.environ["MONGO_URL"]
26
-
27
- client_mongo = MongoClient(MONGO_URL)
28
- db = client_mongo["tiktokbot"]
29
- collection = db["users"]
30
-
31
- RAMDOM_STATUS = [
32
- "civilian",
33
- "wanted",
34
- "undercover",
35
- "rogue_agent",
36
- "innocent",
37
- "fugitive",
38
- "covert_operator"
39
- ]
40
-
41
- def remove_sibyl_system_banned(user_id):
42
- update_doc = {
43
- "sibyl_ban": None,
44
- "reason_sibyl": None,
45
- "is_banned_sibly": None,
46
- "date_joined_sib": None,
47
- "sibyl_userid": None
48
- }
49
- return collection.update_one({"user_id": user_id}, {"$unset": update_doc}, upsert=True)
50
-
51
- def new_sibyl_system_banned(user_id, name, reason, date_joined):
52
- update_doc = {
53
- "sibyl_ban": name,
54
- "reason_sibyl": reason,
55
- "is_banned_sibly": True,
56
- "date_joined_sib": date_joined,
57
- "sibyl_userid": user_id
58
- }
59
- return collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
60
-
61
- def get_sibyl_system_banned(user_id):
62
- user = collection.find_one({"user_id": user_id})
63
- if user:
64
- sibyl_name = user.get("sibyl_ban")
65
- reason = user.get("reason_sibyl")
66
- is_banned = user.get("is_banned_sibly")
67
- date_joined = user.get("date_joined_sib")
68
- sibyl_user_id = user.get("sibyl_userid")
69
- return sibyl_name, reason, is_banned, date_joined, sibyl_user_id
70
- else:
71
- return None
72
-
73
- def get_all_banned():
74
- banned_users = []
75
-
76
- users = collection.find({})
77
-
78
- for user_id in users:
79
- reason = user_id.get("reason_sibyl")
80
- user_id = user_id.get("sibyl_userid")
81
- banned_users.append({"user_id": user_id, "reason": reason})
82
- return banned_users
83
-
84
- def get_all_api_keys():
85
- user = collection.find({})
86
- api_keys = []
87
- for x in user:
88
- api_key = x.get("ryuzaki_api_key")
89
- if api_key:
90
- api_keys.append(api_key)
91
- return api_keys
92
-
93
- def new_profile_clone(
94
- user_id,
95
- first_name,
96
- last_name=None,
97
- profile_id=None,
98
- bio=None
99
- ):
100
- update_doc = {
101
- "first_name_2": first_name,
102
- "last_name_2": last_name,
103
- "profile_id_2": profile_id,
104
- "bio_2": bio
105
- }
106
- collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
107
-
108
- def get_profile_clone(user_id):
109
- user = collection.find_one({"user_id": user_id})
110
- if user:
111
- first_name = user.get("first_name_2")
112
- last_name = user.get("last_name_2")
113
- profile_id = user.get("profile_id_2")
114
- bio = user.get("bio_2")
115
- return first_name, last_name, profile_id, bio
116
- else:
117
- return None, None, None, None