def run(args):
from pajbot.tbutil import load_config
config = load_config(args.config)
if 'main' not in config:
log.error('Missing section [main] in config')
sys.exit(1)
if 'sql' in config:
log.error(
'The [sql] section in config is no longer used. See config.example.ini for the new format under [main].'
)
sys.exit(1)
if 'db' not in config['main']:
log.error('Missing required db config in the [main] section.')
sys.exit(1)
pajbot = Bot(config, args)
pajbot.connect()
def on_sigterm(signal, frame):
pajbot.quit()
sys.exit(0)
signal.signal(signal.SIGTERM, on_sigterm)
try:
pajbot.start()
except KeyboardInterrupt:
pajbot.quit()
pass
def run(args):
from pajbot.tbutil import load_config
config = load_config(args.config)
if 'main' not in config:
log.error('Missing section [main] in config')
sys.exit(1)
if 'sql' in config:
log.error('The [sql] section in config is no longer used. See config.example.ini for the new format under [main].')
sys.exit(1)
if 'db' not in config['main']:
log.error('Missing required db config in the [main] section.')
sys.exit(1)
pajbot = Bot(config, args)
pajbot.connect()
def on_sigterm(signal, frame):
pajbot.quit_bot()
sys.exit(0)
signal.signal(signal.SIGTERM, on_sigterm)
try:
pajbot.start()
except KeyboardInterrupt:
pajbot.quit_bot()
pass
def setUp(self):
from pajbot.bot import Bot
from pajbot.tbutil import load_config
import datetime
config = load_config('config.ini')
args = Bot.parse_args()
self.pajbot = Bot(config, args)
self.source = self.pajbot.users['omgthisuserdoesnotexist123']
self.source.username_raw = 'OmgthisuserdoesnotexiSt123'
self.source.points = 142
self.source.last_seen = datetime.datetime.strptime('17:01:42', '%H:%M:%S')
def setUp(self):
from pajbot.bot import Bot
from pajbot.models.user import User, UserManager
from pajbot.tbutil import load_config
import datetime
config = load_config('config.ini')
args = Bot.parse_args()
self.pajbot = Bot(config, args)
self.source = self.pajbot.users['testuser123Kappa']
self.source.username_raw = 'PajladA'
self.source.points = 142
self.source.last_seen = datetime.datetime.strptime('17:01:42', '%H:%M:%S')
def setUp(self):
from pajbot.bot import Bot
from pajbot.models.user import User, UserManager
from pajbot.tbutil import load_config
import datetime
config = load_config('config.ini')
args = Bot.parse_args()
self.pajbot = Bot(config, args)
self.source = self.pajbot.users['testuser123Kappa']
self.source.username_raw = 'PajladA'
self.source.points = 142
self.source.last_seen = datetime.datetime.strptime(
'17:01:42', '%H:%M:%S')
tag = context.get_tag_argument()
parser = argparse.ArgumentParser()
parser.add_argument('--config',
'-c',
default='config.ini',
help='Specify which config file to use '
'(default: config.ini)')
custom_args = None
if tag is not None:
custom_args = tag.replace('"', '').split()
args, unknown = parser.parse_known_args(args=custom_args)
print('Loading config from {0}'.format(args.config))
tb_config = load_config(args.config)
from pajbot.models.db import Base
# from pajbot.models.user import User
# from pajbot.models.command import Command
# from pajbot.models import *
# from pajbot.tbutil import load_config
# this is the Alembic Config object, which provides
# access to the values within the .ini file in use.
config = context.config
config.set_main_option('sqlalchemy.url', tb_config['main']['db'])
# Interpret the config file for Python logging.
output='scripts/notifications/base.gen.%(version)s.js')
assets.register('notifications_base', notifications_base)
config = configparser.ConfigParser()
parser = argparse.ArgumentParser(description='start the web app')
parser.add_argument('--config', default='config.ini')
parser.add_argument('--host', default='0.0.0.0')
parser.add_argument('--port', type=int, default=2325)
parser.add_argument('--debug', dest='debug', action='store_true')
parser.add_argument('--no-debug', dest='debug', action='store_false')
parser.set_defaults(debug=False)
args = parser.parse_args()
config = load_config(args.config)
if 'web' not in config:
log.error('Missing [web] section in config.ini')
sys.exit(1)
if 'pleblist_password_salt' not in config['web']:
salt = generate_random_salt()
config.set('web', 'pleblist_password_salt', salt.decode('utf-8'))
if 'secret_key' not in config['web']:
salt = generate_random_salt()
config.set('web', 'secret_key', salt.decode('utf-8'))
if 'logo' not in config['web']:
twitchapi = TwitchAPI()
def init(args):
import configparser
import datetime
import logging
import subprocess
import sys
from flask import request
from flask import session
from flask.ext.scrypt import generate_random_salt
import pajbot.web.common
import pajbot.web.routes
from pajbot.bot import Bot
from pajbot.managers.db import DBManager
from pajbot.managers.redis import RedisManager
from pajbot.managers.time import TimeManager
from pajbot.models.module import ModuleManager
from pajbot.models.sock import SocketClientManager
from pajbot.streamhelper import StreamHelper
from pajbot.tbutil import load_config
from pajbot.web.models import errors
from pajbot.web.utils import download_logo
log = logging.getLogger(__name__)
log.info('XD')
config = configparser.ConfigParser()
config = load_config(args.config)
config.read('webconfig.ini')
if 'web' not in config:
log.error('Missing [web] section in config.ini')
sys.exit(1)
if 'pleblist_password_salt' not in config['web']:
salt = generate_random_salt()
config.set('web', 'pleblist_password_salt', salt.decode('utf-8'))
if 'secret_key' not in config['web']:
salt = generate_random_salt()
config.set('web', 'secret_key', salt.decode('utf-8'))
if 'logo' not in config['web']:
res = download_logo(config['main']['streamer'])
if res:
config.set('web', 'logo', 'set')
StreamHelper.init_web(config['main']['streamer'])
redis_options = {}
if 'redis' in config:
redis_options = config._sections['redis']
RedisManager.init(**redis_options)
with open(args.config, 'w') as configfile:
config.write(configfile)
app.bot_modules = config['web'].get('modules', '').split()
app.bot_commands_list = []
app.bot_config = config
app.secret_key = config['web']['secret_key']
if 'sock' in config and 'sock_file' in config['sock']:
SocketClientManager.init(config['sock']['sock_file'])
DBManager.init(config['main']['db'])
TimeManager.init_timezone(config['main'].get('timezone', 'UTC'))
app.module_manager = ModuleManager(None).load()
pajbot.web.routes.admin.init(app)
pajbot.web.routes.api.init(app)
pajbot.web.routes.base.init(app)
pajbot.web.common.filters.init(app)
pajbot.web.common.assets.init(app)
pajbot.web.common.tasks.init(app)
pajbot.web.common.menu.init(app)
app.register_blueprint(pajbot.web.routes.clr.page)
errors.init(app)
pajbot.web.routes.clr.config = config
version = Bot.version
last_commit = ''
commit_number = 0
try:
current_branch = subprocess.check_output(['git', 'rev-parse', '--abbrev-ref', 'HEAD']).decode('utf8').strip()
latest_commit = subprocess.check_output(['git', 'rev-parse', 'HEAD']).decode('utf8').strip()[:8]
commit_number = subprocess.check_output(['git', 'rev-list', 'HEAD', '--count']).decode('utf8').strip()
last_commit = subprocess.check_output(['git', 'log', '-1', '--format=%cd']).decode('utf8').strip()
version = '{0} DEV ({1}, {2}, commit {3})'.format(version, current_branch, latest_commit, commit_number)
except:
pass
default_variables = {
'version': version,
'last_commit': last_commit,
'commit_number': commit_number,
'bot': {
'name': config['main']['nickname'],
},
'site': {
'domain': config['web']['domain'],
'deck_tab_images': config.getboolean('web', 'deck_tab_images'),
'websocket': {
'host': config['websocket'].get('host', config['web']['domain']),
'port': config['websocket']['port'],
'ssl': config.getboolean('websocket', 'ssl')
}
},
'streamer': {
'name': config['web']['streamer_name'],
'full_name': config['main']['streamer']
},
'modules': app.bot_modules,
'request': request,
'session': session,
'google_analytics': config['web'].get('google_analytics', None),
}
if 'streamtip' in config:
default_variables['streamtip_data'] = {
'client_id': config['streamtip']['client_id'],
'redirect_uri': config['streamtip']['redirect_uri'],
}
else:
default_variables['streamtip_data'] = {
'client_id': 'MISSING',
'redirect_uri': 'MISSING',
}
if 'twitchalerts' in config:
default_variables['twitchalerts_data'] = {
'client_id': config['twitchalerts']['client_id'],
'redirect_uri': config['twitchalerts']['redirect_uri'],
}
else:
default_variables['twitchalerts_data'] = {
'client_id': 'MISSING',
'redirect_uri': 'MISSING',
}
@app.context_processor
def current_time():
current_time = {}
current_time['current_time'] = datetime.datetime.now()
return current_time
@app.context_processor
def inject_default_variables():
return default_variables
