96 lines
3.4 KiB
Python
96 lines
3.4 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)
|
|
self.input('prefix', 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')
|