def __init__(self, tls_context, db_operator: DBOperator):
self._db_operator = db_operator
self._rm = ResourceManager()
self._host = "0.0.0.0"
self._port = 5000 # initiate port no above 1024
self._secure_port = 5000
self._stop = False
self._socket = None
self._secure_socket = None
self._tls_context = tls_context
self._logger = LoggingServer.getInstance("overseer")
self._latest_states = {}
self._strategies = {
ServerState.ACCEPT: self._accept_connection,
ServerState.DISPATCH: self._dispatch_connection
}
self._state = ServerState.ACCEPT
self._connection_to_dispatch = None
self._heartbeat_message_interval = 10
self._heartbeat_interval_counters = {}
def __init__(self, updater, devnullChatId=-1):
self._updater = updater
self._resourceManager = ResourceManager()
self._userOptionsManager = UserOptionsManager()
self._usersManager = UsersManager()
self._preambleManager = PreambleManager(self._resourceManager)
self._latexConverter = LatexConverter(self._preambleManager, self._userOptionsManager)
self._inlineQueryResponseDispatcher = InlineQueryResponseDispatcher(updater.bot, self._latexConverter, self._resourceManager, self._userOptionsManager, devnullChatId)
self._messageQueryResponseDispatcher = MessageQueryResponseDispatcher(updater.bot, self._latexConverter, self._resourceManager)
self._devnullChatId = devnullChatId
self._messageFilters = []
self._updater.dispatcher.add_handler(CommandHandler('start', self.onStart))
self._updater.dispatcher.add_handler(CommandHandler('abort', self.onAbort))
self._updater.dispatcher.add_handler(CommandHandler("help", self.onHelp))
self._updater.dispatcher.add_handler(CommandHandler('setcustompreamble', self.onSetCustomPreamble))
self._updater.dispatcher.add_handler(CommandHandler('getmypreamble', self.onGetMyPreamble))
self._updater.dispatcher.add_handler(CommandHandler('getdefaultpreamble', self.onGetDefaultPreamble))
self._updater.dispatcher.add_handler(CommandHandler('setcodeincaptionon', self.onSetCodeInCaptionOn))
self._updater.dispatcher.add_handler(CommandHandler('setcodeincaptionoff', self.onSetCodeInCaptionOff))
self._updater.dispatcher.add_handler(CommandHandler("setdpi", self.onSetDpi))
self._updater.dispatcher.add_handler(MessageHandler(Filters.text, self.dispatchTextMessage), 1)
self._messageFilters.append(self.filterPreamble)
self._messageFilters.append(self.filterExpression)
inline_handler = InlineQueryHandler(self.onInlineQuery)
self._updater.dispatcher.add_handler(inline_handler)
self._usersRequestedCustomPreambleRegistration = set()
def __init__(self, broadcaster, db_operator: DBOperator):
"""
:type broadcaster: src.Broadcaster.Broadcaster
"""
self._broadcaster = broadcaster
self._db_operator = db_operator
self._updater = broadcaster.get_telegram_updater()
self._resource_manager = ResourceManager()
self._logger = LoggingServer.getInstance("overseer")
self._updater.dispatcher.add_handler(CommandHandler('start', self.on_start))
self._updater.dispatcher.add_handler(CommandHandler('help', self.on_help))
self._updater.dispatcher.add_handler(CommandHandler('scheme', self.on_scheme))
self._updater.dispatcher.add_handler(CommandHandler('checkout', self.on_checkout))
self._updater.dispatcher.add_handler(CommandHandler('subscribe', self.on_subscribe))
self._updater.dispatcher.add_handler(CommandHandler('unsubscribe', self.on_unsubscribe))
self._updater.dispatcher.add_handler(CommandHandler('register_slave', self.on_register_slave))
self._updater.dispatcher.add_handler(CommandHandler('abort', self.on_abort))
self._updater.dispatcher.add_handler(MessageHandler(Filters.text, callback=self.on_message))
self._updater.dispatcher.add_handler(CallbackQueryHandler(self.on_callback))
self._message_filters = [self._filter_slave_registration]
self._slave_registration_conversations = {}
self._slave_registration_data = {}
self._slave_registration_functions = {SlaveRegistrationStages.SLAVE_NAME: self._register_slave_name,
SlaveRegistrationStages.SLAVE_PASS: self._register_slave_password}
def setUp(self):
self._conn = psycopg2.connect(
"dbname=%s user=%s password=%s" %
("overseer_test", "inlatexbot", "inlatexbot"))
self._sut = DBOperator("overseer_test",
"inlatexbot",
"inlatexbot",
drop_key="r4jYi1@")
self._rm = ResourceManager()
def __init__(self, telegram_updater: Updater, update_server,
db_operator: DBOperator):
"""
:type update_server: src.UpdateServer.UpdateServer
"""
self._logger = LoggingServer.getInstance("overseer")
self._stop = False
self._telegram_updater = telegram_updater
self._update_server = update_server
self._db_operator = db_operator
self._resource_manager = ResourceManager()
self._running = False
self._executor = ThreadPoolExecutor(max_workers=32)
def __init__(self, dbname, user, password, drop_key=""):
self._dbname = dbname
self._user = user
self._password = password
self._conn = psycopg2.connect("dbname=%s user=%s password=%s" %
(dbname, user, password))
self._c = self._conn.cursor()
if drop_key == "r4jYi1@" and dbname == "overseer_test":
for table in self.TABLES:
try:
with self._conn:
self._c.execute("DROP TABLE %s CASCADE;" % table)
except ProgrammingError as e:
print(e)
if len(self.get_tables()) == 0:
self.create_tables()
self._rm = ResourceManager()
def setUp(self):
userOptionsManager = Mock()
userOptionsManager.getDpiOption = Mock(return_value=720)
self.sut = LatexConverter(PreambleManager(ResourceManager()),
userOptionsManager)
def setUp(self):
self.latexConverter = Mock()
self.bot = Mock()
self.sut = InlineQueryResponseDispatcher(self.bot, self.latexConverter,
ResourceManager(),
UserOptionsManager(), -1)
def setUp(self):
self.latexConverter = Mock()
self.bot = Mock()
self.sut = MessageQueryResponseDispatcher(self.bot,
self.latexConverter,
ResourceManager())
def setUp(self):
self.resourceManager = ResourceManager()
self.sut = PreambleManager(self.resourceManager)
def setUp(self):
self.sut = ResourceManager()