1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
|
"""Legacy URLs."""
# This module reflects the URLs and behavior of the former Django
# application.
import logging
import os
import time
from google.appengine.api import memcache
from flask import Blueprint, make_response, render_template, Response, request
from pygments import highlight
from pygments.formatters import HtmlFormatter
from pygments.lexers import SqlLexer, PythonLexer, PhpLexer
import simplejson as json
import sqlparse
legacy = Blueprint('', 'legacy')
EXAMPLES_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), '../examples'))
@legacy.route('/', methods=['POST', 'GET'])
def index():
data = {'examples': _get_examples()}
extra = {'highlight': True, 'comments': False,
'keywords': 'upper', 'idcase': '',
'n_indents': '2',
'lang': 'sql'}
sql_orig = 'select * from foo join bar on val1 = val2 where id = 123;'
if request.method == 'POST':
oformat = request.form.get('format', 'html')
extra['highlight'] = 'highlight' in request.form
extra['comments'] = 'remove_comments' in request.form
extra['keywords'] = request.form.get('keyword_case', '')
extra['idcase'] = request.form.get('identifier_case', '')
extra['n_indents'] = request.form.get('n_indents', '2')
extra['lang'] = request.form.get('output_format', 'sql')
sql = _get_sql(request.form, request.files)
sql_orig = sql
start = time.time()
data['output'] = _format_sql(sql, request.form, format=oformat)
data['proc_time'] = '%.3f' % (time.time()-start)
if oformat == 'json':
data['errors'] = ''
return make_response(Response(json.dumps(data),
content_type='text/x-json'))
elif oformat == 'text':
return make_response(Response(data['output'], content_type='text/plain'))
data['sql_orig'] = sql_orig
data['extra'] = extra
return render_template('index.html', **data)
@legacy.route('/source/')
def source():
return render_template('source.html')
@legacy.route('/about/')
def about():
return render_template('about.html')
@legacy.route('/api/')
def api():
return render_template('api.html')
@legacy.route('/format/', methods=['GET', 'POST'])
@legacy.route('/format', methods=['GET', 'POST'])
def format_():
if request.method == 'POST':
sql = _get_sql(request.form, request.files)
data = request.form
else:
sql = _get_sql(request.args)
data = request.args
formatted = _format_sql(sql, data, format='text')
return make_response(Response(formatted, content_type='text/plain'))
@legacy.route('/load_example', methods=['GET', 'POST'])
def load_example():
fname = request.form.get('fname')
if fname is None:
answer = 'Uups, I\'ve got no filename...'
elif fname not in _get_examples():
answer = 'Hmm, I think you don\'t want to do that.'
else:
answer = open(os.path.join(EXAMPLES_DIR, fname)).read()
data = json.dumps({'answer': answer})
return make_response(Response(data, content_type='text/x-json'))
def _get_examples():
examples = memcache.get('legacy_examples')
if examples is None:
examples = os.listdir(EXAMPLES_DIR)
memcache.set('legacy_examples', examples)
return examples
def _get_sql(data, files=None):
sql = None
if files is not None and 'datafile' in files:
raw = files['datafile'].read()
try:
sql = raw.decode('utf-8')
except UnicodeDecodeError, err:
logging.error(err)
logging.debug(repr(raw))
sql = (u'-- UnicodeDecodeError: %s\n'
u'-- Please make sure to upload UTF-8 encoded data for now.\n'
u'-- If you want to help improving this part of the application\n'
u'-- please file a bug with some demo data at:\n'
u'-- http://code.google.com/p/python-sqlparse/issues/entry\n'
u'-- Thanks!\n' % err)
if not sql:
sql = data.get('data')
return sql or ''
def _format_sql(sql, data, format='html'):
popts = {}
if data.get('remove_comments'):
popts['strip_comments'] = True
if data.get('keyword_case', 'undefined') not in ('undefined', ''):
popts['keyword_case'] = data.get('keyword_case')
if data.get('identifier_case', 'undefined') not in ('undefined', ''):
popts['identifier_case'] = data.get('identifier_case')
if data.get('n_indents', None) is not None:
val = data.get('n_indents')
try:
popts['indent_width'] = max(1, min(1000, int(val)))
popts['reindent'] = True
except (ValueError, TypeError):
pass
if (not 'indent_width' in popts and
data.get('reindent', '').lower() in ('1', 'true', 't')):
popts['indent_width'] = 2
popts['reindent'] = True
if data.get('output_format', None) is not None:
popts['output_format'] = data.get('output_format')
logging.debug('Format: %s, POPTS: %r', format, popts)
logging.debug(sql)
sql = sqlparse.format(sql, **popts)
if format in ('html', 'json'):
if data.get('highlight', False):
if popts['output_format'] == 'python':
lexer = PythonLexer()
elif popts['output_format'] == 'php':
lexer = PhpLexer()
else:
lexer = SqlLexer()
sql = highlight(sql, lexer, HtmlFormatter())
else:
sql = ('<textarea class="resizable" '
'style="height: 350px; margin-top: 1em;">%s</textarea>'
% sql)
return sql
|
