2017-11-17 14:56:38 +00:00
|
|
|
import codecs
|
2017-11-27 17:52:34 +00:00
|
|
|
import sqlite3
|
2018-01-10 15:49:23 +00:00
|
|
|
from functools import partial
|
2017-11-17 14:56:38 +00:00
|
|
|
|
2017-11-27 17:52:34 +00:00
|
|
|
import source_code
|
2018-01-17 13:26:16 +00:00
|
|
|
from event_handler_base import EventHandlerBase
|
2018-01-10 15:49:23 +00:00
|
|
|
from source_code_event_handler import SourceCodeEventHandler
|
2017-11-17 14:56:38 +00:00
|
|
|
from tornado.ioloop import IOLoop
|
|
|
|
|
2017-11-27 17:52:34 +00:00
|
|
|
from login_component import authorize_login
|
2018-01-10 15:49:23 +00:00
|
|
|
from util import create_source_code_response_data
|
|
|
|
|
|
|
|
login_component_py_response = partial(
|
|
|
|
create_source_code_response_data, filename='login_component.py', language='python'
|
|
|
|
)
|
2017-11-17 14:56:38 +00:00
|
|
|
|
|
|
|
|
2018-01-17 13:26:16 +00:00
|
|
|
class EchoHandler(EventHandlerBase):
|
|
|
|
def handle_event(self, anchor, data_json):
|
|
|
|
return data_json
|
2017-11-17 14:56:38 +00:00
|
|
|
|
|
|
|
|
2018-01-17 13:26:16 +00:00
|
|
|
class Rot13Handler(EventHandlerBase):
|
|
|
|
def handle_event(self, anchor, data_json):
|
|
|
|
data_json['data'] = codecs.encode(data_json['data'], 'rot13')
|
|
|
|
return data_json
|
2017-11-17 14:56:38 +00:00
|
|
|
|
|
|
|
|
2018-01-17 13:26:16 +00:00
|
|
|
class ChangeCaseHandler(EventHandlerBase):
|
|
|
|
def handle_event(self, anchor, data_json):
|
|
|
|
data_json['data'] = data_json['data'].upper() if data_json['data'].islower() else data_json['data'].lower()
|
|
|
|
return data_json
|
2017-11-17 14:56:38 +00:00
|
|
|
|
|
|
|
|
2018-01-17 13:26:16 +00:00
|
|
|
class ReverseHandler(EventHandlerBase):
|
|
|
|
def handle_event(self, anchor, data_json):
|
|
|
|
data_json['data'] = data_json['data'][::-1]
|
|
|
|
return data_json
|
2017-11-17 14:56:38 +00:00
|
|
|
|
|
|
|
|
2018-01-17 13:26:16 +00:00
|
|
|
class LoginHandler(EventHandlerBase):
|
|
|
|
def handle_event(self, anchor, data_json):
|
|
|
|
email, password = data_json['data']['email'], data_json['data']['password']
|
|
|
|
try:
|
|
|
|
sql_statement = source_code.find_local_variable_value(authorize_login, 'sql_statement')
|
|
|
|
yield (
|
|
|
|
'anchor_logger',
|
|
|
|
'The SQL statement executed by the server will look like this:\n `{}`'.format(sql_statement)
|
|
|
|
)
|
2017-11-27 17:52:34 +00:00
|
|
|
|
2018-01-17 13:26:16 +00:00
|
|
|
yield ('anchor_webide',
|
|
|
|
login_component_py_response(content=source_code.get_source_code(authorize_login, strip_comments=False)))
|
2017-11-27 17:52:34 +00:00
|
|
|
|
2018-01-17 13:26:16 +00:00
|
|
|
sql_statement_with_values = sql_statement.format(email, password)
|
|
|
|
yield (
|
|
|
|
'anchor_logger',
|
|
|
|
'After the submitted parameters are substituted it looks like this:\n `{}`'.format(
|
|
|
|
sql_statement_with_values
|
|
|
|
)
|
2017-11-27 17:52:34 +00:00
|
|
|
)
|
|
|
|
|
2018-01-17 13:26:16 +00:00
|
|
|
logged_in_email, is_admin = authorize_login(email, password)
|
2017-11-27 17:52:34 +00:00
|
|
|
|
2018-01-17 13:26:16 +00:00
|
|
|
yield (
|
|
|
|
'anchor_logger',
|
|
|
|
'After the query is executed, it returns _{}_ as email address, and _{}_ for is_admin'.format(
|
|
|
|
logged_in_email, is_admin
|
|
|
|
)
|
2017-11-27 17:52:34 +00:00
|
|
|
)
|
|
|
|
|
2018-01-17 13:26:16 +00:00
|
|
|
if logged_in_email is not None:
|
|
|
|
response = 'Logged in as _{}_. You __{}have__ admin privileges.'.format(
|
|
|
|
logged_in_email,
|
|
|
|
'' if is_admin else 'don\'t '
|
|
|
|
)
|
|
|
|
else:
|
|
|
|
response = 'Bad username/password!'
|
|
|
|
except sqlite3.Warning:
|
|
|
|
response = 'Invalid request!'
|
2017-11-27 17:52:34 +00:00
|
|
|
|
2018-01-17 13:26:16 +00:00
|
|
|
yield ('anchor_login', '# Login page\n' + response)
|
2017-11-27 17:52:34 +00:00
|
|
|
|
|
|
|
|
2017-11-17 14:56:38 +00:00
|
|
|
if __name__ == '__main__':
|
2018-01-17 13:26:16 +00:00
|
|
|
anchor_a = ChangeCaseHandler('anchor_a')
|
|
|
|
anchor_b = Rot13Handler('anchor_b')
|
|
|
|
anchor_c = ReverseHandler('anchor_c')
|
2018-01-10 15:49:23 +00:00
|
|
|
anchor_webide = SourceCodeEventHandler('anchor_webide', 'login_component.py', 'login')
|
2017-11-17 14:56:38 +00:00
|
|
|
IOLoop.instance().start()
|
2017-11-27 17:52:34 +00:00
|
|
|
|