-
Notifications
You must be signed in to change notification settings - Fork 0
/
definitions.py
158 lines (138 loc) · 7.22 KB
/
definitions.py
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
# New encryption method (1.6)!
from tinydb import TinyDB, Query, where
from tinydb.operations import add, subtract, delete
import tinydb_encrypted_jsonstorage as tae
from os.path import join as join_path
from os.path import exists
from random import *
import string
import sys
import os
# Boolean to string parser.
from distutils.util import strtobool
def createConfigFile():
print("We haven't detected a config file! Let's do some setup before we get started.\n")
if not exists("json/"):
os.mkdir('json')
cfg = TinyDB("json/config.json") #Creates the config file.
# Encryption key
print("Enter the ENCRYPTION KEY (1/11)")
print("The bot automatically encrypts its databases, so not you nor anyone else can access them. This is the encryption key the bot uses to encrypt said databases. This string can be anything, but it's recommended to use a strong, randomly-chosen password, just in case. DO NOT SHARE THIS KEY.")
# Generate a secure password.
characters = string.ascii_letters + string.punctuation + string.digits
randomPassword = "".join(choice(characters) for x in range(randint(16, 32)))
print("(default: " + randomPassword + ")")
pwdInput = input("")
if not pwdInput:
pwdInput = randomPassword
# Bot token
print("\nEnter BOT TOKEN (2/11)")
print("This is the bot token Discord generates for you when you've created a bot on the Discord Developer Portal. Here's an in-depth guide on how to create a Discord bot, if you haven't done so already. DO NOT SHARE THIS TOKEN.")
tknInput = input("")
if not tknInput:
sys.exit()
# Bot name
print("\nEnter BOT NAME (3/11)")
print("Replaces all instances of the bot's name (in this case, Reto) to its own. Change this if you'll name your bot anything other than Reto.")
print("(default: Reto)")
nmeInput = input("")
if not nmeInput:
nmeInput = "Reto"
# Prefix
print("\nEnter PREFIX (4/11):")
print("The default prefix Reto uses on DMs and all servers that haven't set their own custom prefix.")
print("(default: ?)")
prxInput = input("")
if not prxInput:
prxInput = "?"
# Support server
print("\nEnter SUPPORT SERVER (5/11):")
print("Links to the support server shown on the ?setup command, error messages, et cetera.")
print("(default: https://discord.gg/rrszpTN)")
ssvInput = input("")
if not ssvInput:
ssvInput = "https://discord.gg/rrszpTN"
# Bot version
print("\nEnter BOT VERSION (6/11):")
print("The bot's current version. Appears on its Activities and other places.")
print("(default: 1.6c)")
verInput = input("")
if not verInput:
verInput = "1.6c"
# Github Username
print("\nEnter GITHUB USERNAME (7/11):")
print("If you're hosting Reto on Github, drop your username here! This will help it keep track of updates and inform you if you've got to update it.\nIf you haven't got one (or would rather be informed of the main Reto branch's updates), leave as default.")
print("(default: despedite)")
gitInput = input("")
if not gitInput:
gitInput = "despedite"
# Github Username
print("\nEnter GITHUB REPOSITORY (8/11):")
print("If you're hosting Reto on Github, drop your repository name here! This will help it keep track of updates and inform you if you've got to update it.\nIf you haven't got one (or would rather be informed of the main Reto branch's updates), leave as default.")
print("(default: reto)")
rpoInput = input("")
if not rpoInput:
rpoInput = "reto"
# Bot owner
print("\nEnter BOT OWNER (9/11):")
print("An array of the people that are the bot's developers. You should fill this out with your user ID to gain access to certain bot owner specific commands (and a badge on your ?profile).")
print("(enter IDs one after the other, press ENTER on an empty new line when you're done)")
ownInput = []
ownInput.append(input(""))
if not ownInput[0]:
sys.exit()
while ownInput[len(ownInput) - 1]:
ownInput.append(input(""))
ownInput.remove(ownInput[len(ownInput)-1])
# Ephemeral
print("\nEnter EPHEMERAL STORAGE (10/11):")
print("Discord asks for people to delete the posts they have on users after 30 days. Leaving this on means that Reto will periodically check every 12 hours to delete old posts. If this is a private bot, you can turn this off.")
print("(True/False, default/invalid: True)")
ephInput = input("")
if (ephInput!= "True" or "False") or not ephInput:
ephInput = "True"
# Debug
print("\nEnter DEBUG (11/11):")
print("Boolean (True/False) that's useful if you wanna test certain commands alone. Its main use is letting you vote on your own posts. Make sure this is set as 'False' when you're done coding.")
print("(True/False, default/invalid: False)")
dbgInput = input("")
if (dbgInput!= "True" or "False") or not dbgInput:
dbgInput = "False"
# Insert all that data!
cfg.insert({'key': pwdInput, 'bottoken': tknInput, "botname": nmeInput, "prefix": prxInput, "support": ssvInput, "botver": verInput, "githubusername": gitInput, "githubrepo": rpoInput, "botowner": ownInput, "ephemeral": ephInput, "debug": dbgInput})
if not exists("json/") or not exists("json/config.json"):
createConfigFile()
# Unencrypted! This NEEDS to be editable by a simple text editor.
cfg = TinyDB("json/config.json") #Config file: stores configurations for the bot. Modify at your heart's content!
for c in cfg:
bottoken = c['bottoken']
botname = c['botname']
support = c['support']
botver = c['botver']
prefix = c['prefix']
botowner = c['botowner']
key = c['key']
gusername = c['githubusername']
grepo = c['githubrepo']
ephemeral = c['ephemeral']
ephemeral = bool(strtobool(ephemeral)) # Otherwhise, it's a string.
debug = c['debug']
debug = bool(strtobool(debug)) # Otherwhise, it's a string.
db = TinyDB(encryption_key=key, path=join_path("db/","profile.reto"), storage=tae.EncryptedJSONStorage)
#Database file: stores points of every user.
srv = TinyDB(encryption_key=key, path=join_path("db/","srv.reto"), storage=tae.EncryptedJSONStorage)
#Server-specific configuration - allows you to modify stuff like the name of the reactions, for example.
activity = TinyDB(encryption_key=key, path=join_path("db/","activity.reto"), storage=tae.EncryptedJSONStorage)
#Activity file: the "Playing" commands the bot has.
post = TinyDB(encryption_key=key, path=join_path("db/","comments.reto"), storage=tae.EncryptedJSONStorage)
#Comment file: every comment that's been reacted to is saved here.
priv = TinyDB(encryption_key=key, path=join_path("db/","blacklist.reto"), storage=tae.EncryptedJSONStorage)
#Privacy Mode: Users with PM on will not have their messages logged in the comment leaderboard.
best = TinyDB(encryption_key=key, path=join_path("db/","best.reto"), storage=tae.EncryptedJSONStorage)
#Best Of name: Used to look up the Best-Of name of the channel.
dm = TinyDB(encryption_key=key, path=join_path("db/","deletables.reto"), storage=tae.EncryptedJSONStorage)
#Leaderboard logs: Database used to track the messages sent on scrolling leaderboards.
customprefix = TinyDB(encryption_key=key, path=join_path("db/","customprefix.reto"), storage=tae.EncryptedJSONStorage)
#Prefix file: Stores a server's custom prefix.
chan = TinyDB(encryption_key=key, path=join_path("db/","channels.reto"), storage=tae.EncryptedJSONStorage)
#Channel file: Used for Autovote specific channels and other channel features.