tuxbot-bot/first_run/initializer.py

94 lines
3.3 KiB
Python

from pip._internal import main as pip
import shutil
from .langs import locales, texts
class Config:
def __init__(self):
self.config = {
'log_channel_id': '<INSERT_LOG_CHANNEL_HERE (in int)>',
'main_server_id': '<INSERT_MAIN_CHANNEL_ID_HERE (in int)>',
'authorized_id': '[admin ids here (in int)]',
'unkickable_id': '[unkickable ids here (in int)]'
}
with open('requirements.txt', 'r') as f:
self.packages = f.read().split('\n')
def input(self, key, **kwargs):
lang = self.config.get('lang', 'multiple')
print('\n\033[4m' + texts.get(lang).get(key) + '\033[0m')
response = input('> ')
if kwargs.get('valid'):
while response not in kwargs.get('valid'):
print('\033[36m' + '/'.join(kwargs.get('valid')) + '\033[0m')
response = input('> ')
if not kwargs.get('empty', True):
while len(response) == 0:
print('\033[41m' + texts.get(lang).get('not_empty')
+ '\033[0m')
response = input('> ')
else:
response = kwargs.get('default', None) if len(response) == 0 \
else response
self.config[key] = response
def install(self):
self.input('lang', valid=locales)
print('\n\n\033[4;36m'
+ texts.get(self.config.get('lang')).get('install')
+ '\033[0m\n')
for package in self.packages:
pip(['install', package])
def ask(self):
print('\n\n\033[4;36m' + texts.get(self.config.get('lang')).get('conf')
+ '\033[0m\n')
self.input('token', empty=False)
self.input('postgresql_username', empty=False)
self.input('postgresql_password', empty=False)
self.input('postgresql_dbname', empty=False)
print('\n\n\033[4;36m' + texts.get(self.config.get('lang')).get('logs')
+ '\033[0m\n')
self.input('wh_id', empty=True)
self.input('wh_token', empty=True)
print('\n\n\033[4;36m' + texts.get(self.config.get('lang')).get('misc')
+ '\033[0m\n')
self.input('activity', empty=True)
def save(self):
with open('config.py', 'w') as file:
postgresql = f"postgresql://" \
f"{self.config.get('postgresql_username')}:" \
f"{self.config.get('postgresql_password')}" \
f"@localhost/{self.config.get('postgresql_dbname')}"
file.write(f"postgresql = '{postgresql}'\n")
logs_webhook = dict(id=int(self.config.get('wh_id')),
token=self.config.get('wh_token'))
file.write(f"logs_webhook = '{logs_webhook}'\n")
for key, value in self.config.items():
if not key.startswith('postgresql_') \
and not key.startswith('wh_'):
value = f"'{value}'" if type(value) is str else value
file.write(f"{key} = {value}\n")
print('\n\n\033[4;36m' + texts.get(self.config.get('lang')).get('end')
+ '\033[0m\n')
def clean(self):
print('\n\n\033[4;36m'
+ texts.get(self.config.get('lang')).get('clean')
+ '\033[0m\n')
shutil.rmtree('first_run')