def __init__(self,
*args,
poll_cache_file=DEFAULT_POLL_CACHE_FILENAME,
**kwargs):
super().__init__(*args, **kwargs)
self.polls = []
self.polls_by_msgid = {}
self.poll_cache_file = poll_cache_file
self.listeners = {
"on_ready": {},
"finalize": {},
"before_on_message": {},
"after_on_message": {},
"on_reaction_add": {},
"on_reaction_remove": {},
"on_message_delete": {}
}
self.command_handler = commands.CommandHandler(prefix=self.PREFIX)
modules.defaults.inject_module(self)
modules.pollmaker.inject_module(self)
self.ready = False
commands.InsertIntoTable(line, cmd)
elif (line.upper().find("UPDATE ") != -1):
commands.UpdateTable(line, cmd)
elif (line.upper().find("SET ") != -1):
commands.SetTable(line, cmd)
elif ((line.upper().find("DELETE ") != -1)
and (line.upper().find(" FROM ") != -1)):
commands.DeleteFromTable(line, cmd)
elif (line.upper().find("FROM ") != -1):
commands.FromTable(line, cmd)
elif (line.upper().find("WHERE ") != -1):
if (cmd.DataManipulation != commands.cmdName.INNER_JOIN):
commands.WhereTable(line, cmd)
else:
commands.OnTable(line, cmd)
elif (line.upper().find("ON ") != -1):
commands.OnTable(line, cmd)
else:
print("!Invalid command found.")
#Main Function
if __name__ == '__main__':
try:
sqlFile = open(str(sys.argv[1]), "r").read().splitlines()
except:
exit("!No filename found in command line argument.")
cmd = commands.CommandHandler()
for line in sqlFile:
ReadLine(line, cmd)
import time
import json
import commands
import logging
import sys
context = zmq.Context()
control = context.socket(zmq.REP)
publisher = context.socket(zmq.PUB)
control.bind("tcp://*:5555")
publisher.bind("tcp://*:5556")
poller = zmq.Poller()
poller.register(control, zmq.POLLIN)
clients = {}
handler = commands.CommandHandler(clients, logging.Logger(publisher))
while True:
socks = dict(poller.poll(10))
if control in socks and socks[control] == zmq.POLLIN:
command = json.loads(control.recv())
response, events = handler.handle(command)
control.send_string(response)
for event in events:
publisher.send_multipart(event)
if event[0] == "SHUTDOWN":
sys.exit(0)
async def init(self):
self.command_handler = commands.CommandHandler(self)
await self.command_handler.loadConfig()
await self.command_handler.loadModulesFromConfig()
if __name__ == '__main__':
logging.basicConfig(filename="./log.log", level=logging.INFO)
global req_dict
logger = logging.getLogger('core')
cmd_queue = cmd_queue.CommandQueue()
b = bot.Bot.create_bot_from_config()
while True:
try:
event = b.keep_alive_loop()
if event is not None:
ch = commands.CommandHandler(event)
if ch.check_for_command_character():
req_dict = ch.parse_msg_to_req_dict()
permission = ch.check_permission()
allowed_bool = permission[0]
current_permission_level = permission[1]
needed_permission = permission[2]
req_dict['cmd']['permission'][
'have'] = current_permission_level
req_dict['cmd']['permission']['need'] = needed_permission
print(req_dict)
logger.info(str(req_dict))
if allowed_bool: