logotron_genesis.kv 1 #!/usr/bin/python
logotron_genesis.kv 2
logotron_genesis.kv 3 ##############################################################################
logotron_genesis.kv 4 import ConfigParser, sys
logotron_genesis.kv 5 import psycopg2, psycopg2.extras
logotron_genesis.kv 6 import psycopg2.extensions
logotron_genesis.kv 7 psycopg2.extensions.register_type(psycopg2.extensions.UNICODE)
logotron_genesis.kv 8 psycopg2.extensions.register_type(psycopg2.extensions.UNICODEARRAY)
logotron_genesis.kv 9 import time
logotron_genesis.kv 10 import datetime
logotron_genesis.kv 11 from datetime import timedelta
logotron_genesis.kv 12 import sys
logotron_genesis.kv 13 reload(sys)
logotron_genesis.kv 14 sys.setdefaultencoding('utf8')
logotron_genesis.kv 15 import os
logotron_genesis.kv 16 import threading
logotron_genesis.kv 17 import re
logotron_genesis.kv 18 from datetime import datetime
logotron_genesis.kv 19 from urlparse import urljoin
raw_line_export.kv 20 from flask import Flask, request, session, url_for, redirect, Response, \
logotron_genesis.kv 21 render_template, abort, g, flash, _app_ctx_stack, make_response, \
logotron_genesis.kv 22 jsonify
logotron_genesis.kv 23 from flask import Flask
logotron_genesis.kv 24 from flask.ext.cache import Cache
logotron_genesis.kv 25 ##############################################################################
logotron_genesis.kv 26
logotron_genesis.kv 27 ##############################################################################
logotron_genesis.kv 28 # Single mandatory arg: config file path
logotron_genesis.kv 29 if len(sys.argv[1:]) != 1:
logotron_genesis.kv 30 # If no args, print usage and exit:
logotron_genesis.kv 31 print sys.argv[0] + " CONFIG"
logotron_genesis.kv 32 exit(0)
logotron_genesis.kv 33
logotron_genesis.kv 34 # Read Config from given conf file
logotron_genesis.kv 35 config_path = os.path.abspath(sys.argv[1])
logotron_genesis.kv 36 cfg = ConfigParser.ConfigParser()
logotron_genesis.kv 37 cfg.readfp(open(config_path))
logotron_genesis.kv 38
logotron_genesis.kv 39 try:
logotron_genesis.kv 40 # IRCism:
logotron_genesis.kv 41 Nick = cfg.get("irc", "nick")
logotron_genesis.kv 42 Channels = [x.strip() for x in cfg.get("irc", "chans").split(',')]
logotron_genesis.kv 43 Bots = [x.strip() for x in cfg.get("logotron", "bots").split(',')]
logotron_genesis.kv 44 Bots.append(Nick) # Add our own bot to the bot list
logotron_genesis.kv 45 # DBism:
logotron_genesis.kv 46 DB_Name = cfg.get("db", "db_name")
logotron_genesis.kv 47 DB_User = cfg.get("db", "db_user")
raw_line_export.kv 48 DB_DEBUG = int(cfg.get("db", "db_debug"))
logotron_genesis.kv 49 # Logism:
logotron_genesis.kv 50 Base_URL = cfg.get("logotron", "base_url")
logotron_genesis.kv 51 Era = int(cfg.get("logotron", "era"))
raw_line_export.kv 52 DEBUG = int(cfg.get("logotron", "www_dbg"))
raw_line_export.kv 53 Max_Raw_Ln = int(cfg.get("logotron", "max_raw"))
logotron_genesis.kv 54 # WWW:
logotron_genesis.kv 55 WWW_Port = int(cfg.get("logotron", "www_port"))
logotron_genesis.kv 56
logotron_genesis.kv 57 except Exception as e:
logotron_genesis.kv 58 print "Invalid config: ", e
logotron_genesis.kv 59 exit(1)
logotron_genesis.kv 60
logotron_genesis.kv 61 ##############################################################################
logotron_genesis.kv 62
logotron_genesis.kv 63 ##############################################################################
logotron_genesis.kv 64 ### Knobs not made into config yet ###
logotron_genesis.kv 65 Default_Chan = Channels[0]
logotron_genesis.kv 66 Min_Query_Length = 3
logotron_genesis.kv 67 Max_Search_Results = 1000
logotron_genesis.kv 68
logotron_genesis.kv 69 ## Format for Date in Log Lines
logotron_genesis.kv 70 Date_Short_Format = "%Y-%m-%d"
logotron_genesis.kv 71 ##############################################################################
logotron_genesis.kv 72
logotron_genesis.kv 73 app = Flask(__name__)
logotron_genesis.kv 74 cache = Cache(app,config={'CACHE_TYPE': 'simple'})
logotron_genesis.kv 75 app.config.from_object(__name__)
logotron_genesis.kv 76
logotron_genesis.kv 77 def get_db():
logotron_genesis.kv 78 db = getattr(g, 'db', None)
logotron_genesis.kv 79 if db is None:
logotron_genesis.kv 80 db = g.db = psycopg2.connect("dbname=%s user=%s" % (DB_Name, DB_User))
logotron_genesis.kv 81 return db
logotron_genesis.kv 82
logotron_genesis.kv 83 def close_db():
logotron_genesis.kv 84 if hasattr(g, 'db'):
logotron_genesis.kv 85 g.db.close()
logotron_genesis.kv 86
logotron_genesis.kv 87 @app.before_request
logotron_genesis.kv 88 def before_request():
logotron_genesis.kv 89 g.db = get_db()
logotron_genesis.kv 90
logotron_genesis.kv 91 @app.teardown_request
logotron_genesis.kv 92 def teardown_request(exception):
logotron_genesis.kv 93 close_db()
logotron_genesis.kv 94
logotron_genesis.kv 95 def query_db(query, args=(), one=False):
logotron_genesis.kv 96 cur = get_db().cursor(cursor_factory=psycopg2.extras.RealDictCursor)
logotron_genesis.kv 97 if (DB_DEBUG): print "query: '{0}'".format(query)
logotron_genesis.kv 98 cur.execute(query, args)
logotron_genesis.kv 99 rv = cur.fetchone() if one else cur.fetchall()
logotron_genesis.kv 100 if (DB_DEBUG): print "query res: '{0}'".format(rv)
logotron_genesis.kv 101 return rv
logotron_genesis.kv 102
logotron_genesis.kv 103 def exec_db(query, args=()):
logotron_genesis.kv 104 cur = get_db().cursor(cursor_factory=psycopg2.extras.RealDictCursor)
logotron_genesis.kv 105 if (DB_DEBUG): print "query: '{0}'".format(query)
logotron_genesis.kv 106 if (DB_DEBUG): print "args: '{0}'".format(args)
logotron_genesis.kv 107 if (DB_DEBUG): print "EXEC:"
logotron_genesis.kv 108 cur.execute(query, args)
logotron_genesis.kv 109
logotron_genesis.kv 110 def getlast_db():
logotron_genesis.kv 111 cur = get_db().cursor(cursor_factory=psycopg2.extras.RealDictCursor)
logotron_genesis.kv 112 cur.execute('select lastval()')
logotron_genesis.kv 113 return cur.fetchone()['lastval']
logotron_genesis.kv 114
logotron_genesis.kv 115 def commit_db():
logotron_genesis.kv 116 cur = get_db().cursor(cursor_factory=psycopg2.extras.RealDictCursor)
logotron_genesis.kv 117 g.db.commit()
logotron_genesis.kv 118
logotron_genesis.kv 119 ##############################################################################
logotron_genesis.kv 120
logotron_genesis.kv 121 ## All eggogs redirect to main page
logotron_genesis.kv 122 @app.errorhandler(404)
logotron_genesis.kv 123 def page_not_found(error):
logotron_genesis.kv 124 return redirect(url_for('log'))
logotron_genesis.kv 125
logotron_genesis.kv 126 ##############################################################################
logotron_genesis.kv 127
logotron_genesis.kv 128 html_escape_table = {
logotron_genesis.kv 129 "&": "&",
logotron_genesis.kv 130 '"': """,
logotron_genesis.kv 131 "'": "'",
logotron_genesis.kv 132 ">": ">",
logotron_genesis.kv 133 "<": "<",
logotron_genesis.kv 134 }
logotron_genesis.kv 135
logotron_genesis.kv 136 def html_escape(text):
logotron_genesis.kv 137 return "".join(html_escape_table.get(c,c) for c in text)
logotron_genesis.kv 138
logotron_genesis.kv 139 ##############################################################################
logotron_genesis.kv 140
logotron_genesis.kv 141 ## Get base URL
logotron_genesis.kv 142 def get_base():
logotron_genesis.kv 143 if DEBUG:
logotron_genesis.kv 144 return request.host_url
logotron_genesis.kv 145 return Base_URL
logotron_genesis.kv 146
logotron_genesis.kv 147
logotron_genesis.kv 148 # Get perma-URL corresponding to given log line
logotron_genesis.kv 149 def line_url(l):
logotron_genesis.kv 150 return "{0}log/{1}/{2}#{3}".format(get_base(),
logotron_genesis.kv 151 l['chan'],
logotron_genesis.kv 152 l['t'].strftime(Date_Short_Format),
logotron_genesis.kv 153 l['idx'])
logotron_genesis.kv 154
logotron_genesis.kv 155 def gen_chanlist(selected_chan):
logotron_genesis.kv 156 # Get current time
logotron_genesis.kv 157 now = datetime.now()
logotron_genesis.kv 158
logotron_genesis.kv 159 s = """<table align="center" class="chantable"><tr>"""
logotron_genesis.kv 160 for chan in Channels:
logotron_genesis.kv 161 chan_formed = chan
logotron_genesis.kv 162 if chan == selected_chan:
logotron_genesis.kv 163 chan_formed = "<span class='highlight'>" + chan + "</span>"
logotron_genesis.kv 164 s += """<th><a href="{0}log/{1}">{2}</a></th>""".format(
logotron_genesis.kv 165 get_base(), chan, chan_formed)
logotron_genesis.kv 166 s += "</tr><tr>"
logotron_genesis.kv 167
logotron_genesis.kv 168 for chan in Channels:
logotron_genesis.kv 169
logotron_genesis.kv 170 last_time = query_db(
logotron_genesis.kv 171 '''select t, idx from loglines where chan=%s
logotron_genesis.kv 172 and idx = (select max(idx) from loglines where chan=%s) ;''',
logotron_genesis.kv 173 [chan, chan], one=True)
logotron_genesis.kv 174
logotron_genesis.kv 175 last_time_txt = ""
logotron_genesis.kv 176 if last_time != None:
logotron_genesis.kv 177 span = (now - last_time['t'])
logotron_genesis.kv 178 days = span.days
logotron_genesis.kv 179 hours = span.seconds/3600
logotron_genesis.kv 180 minutes = (span.seconds%3600)/60
logotron_genesis.kv 181
logotron_genesis.kv 182 if days != 0:
logotron_genesis.kv 183 last_time_txt += '%dd ' % days
logotron_genesis.kv 184 if hours != 0:
logotron_genesis.kv 185 last_time_txt += '%dh ' % hours
logotron_genesis.kv 186 if minutes != 0:
logotron_genesis.kv 187 last_time_txt += '%dm' % minutes
logotron_genesis.kv 188
logotron_genesis.kv 189 s += """<td><i><a href="{0}log/{1}/{2}#{3}">{4}</a></i></td>""".format(
logotron_genesis.kv 190 get_base(),
logotron_genesis.kv 191 chan,
logotron_genesis.kv 192 last_time['t'].strftime(Date_Short_Format),
logotron_genesis.kv 193 last_time['idx'],
logotron_genesis.kv 194 last_time_txt)
logotron_genesis.kv 195
logotron_genesis.kv 196 else:
logotron_genesis.kv 197 last_time_txt = ""
logotron_genesis.kv 198 s += "<td></td>"
logotron_genesis.kv 199
logotron_genesis.kv 200 s += "</tr></table>"
logotron_genesis.kv 201 return s
logotron_genesis.kv 202
logotron_genesis.kv 203
logotron_genesis.kv 204 # Make above callable from inside htm templater:
logotron_genesis.kv 205 app.jinja_env.globals.update(gen_chanlist=gen_chanlist)
logotron_genesis.kv 206
logotron_genesis.kv 207
logotron_genesis.kv 208 # HTML Tag Regex
logotron_genesis.kv 209 tag_regex = re.compile("(<[^>]+>)")
logotron_genesis.kv 210
logotron_genesis.kv 211
logotron_genesis.kv 212 # Find the segments of a block of text which constitute HTML tags
logotron_genesis.kv 213 def get_link_intervals(str):
logotron_genesis.kv 214 links = []
logotron_genesis.kv 215 span = []
logotron_genesis.kv 216 for match in tag_regex.finditer(str):
logotron_genesis.kv 217 span = match.span()
logotron_genesis.kv 218 links += [span]
logotron_genesis.kv 219 return links
logotron_genesis.kv 220
logotron_genesis.kv 221
logotron_genesis.kv 222 # Highlight all matched tokens in given text
logotron_genesis.kv 223 def highlight_matches(strings, text):
logotron_genesis.kv 224 e = '(' + ('|'.join(strings)) + ')'
logotron_genesis.kv 225 return re.sub(e,
logotron_genesis.kv 226 r"""<span class='highlight'>\1</span>""",
logotron_genesis.kv 227 text,
logotron_genesis.kv 228 flags=re.I)
logotron_genesis.kv 229
logotron_genesis.kv 230
logotron_genesis.kv 231 # Highlight matched tokens in the display of a search result logline,
logotron_genesis.kv 232 # but leave HTML tags alone
logotron_genesis.kv 233 def highlight_text(strings, text):
logotron_genesis.kv 234 result = ""
logotron_genesis.kv 235 last = 0
logotron_genesis.kv 236 for i in get_link_intervals(text):
logotron_genesis.kv 237 i_start, i_end = i
logotron_genesis.kv 238 result += highlight_matches(strings, text[last:i_start])
logotron_genesis.kv 239 result += text[i_start:i_end] # the HTML tag, leave it alone
logotron_genesis.kv 240 last = i_end
logotron_genesis.kv 241 result += highlight_matches(strings, text[last:]) # last block
logotron_genesis.kv 242 return result
logotron_genesis.kv 243
logotron_genesis.kv 244
logotron_genesis.kv 245 # Regexps used in format_logline:
logotron_genesis.kv 246 boxlinks_re = re.compile('\[\s*<a href="(http[^ \[\]]+)">[^ <]+</a>\s*\]\[([^\[\]]+)\]')
logotron_genesis.kv 247 stdlinks_re = re.compile('(http[^ \[\]]+)')
logotron_genesis.kv 248
logotron_genesis.kv 249
logotron_genesis.kv 250 ## Format given log line for display
multsel_and_datef... 251 def format_logline(l, highlights = [], select=[]):
logotron_genesis.kv 252 payload = html_escape(l['payload'])
logotron_genesis.kv 253
logotron_genesis.kv 254 # Format ordinary links:
logotron_genesis.kv 255 payload = re.sub(stdlinks_re, r'<a href="\1">\1</a>', payload)
logotron_genesis.kv 256
logotron_genesis.kv 257 # Now also format [link][text] links :
logotron_genesis.kv 258 payload = re.sub(boxlinks_re, r'<a href="\1">\2</a>', payload)
logotron_genesis.kv 259
logotron_genesis.kv 260 # If this is a search result, illuminate the matched strings:
logotron_genesis.kv 261 if highlights != []:
logotron_genesis.kv 262 payload = highlight_text(highlights, payload)
logotron_genesis.kv 263
logotron_genesis.kv 264 bot = ""
logotron_genesis.kv 265 if l['speaker'] in Bots:
logotron_genesis.kv 266 bot = " bot"
logotron_genesis.kv 267
multsel_and_datef... 268 # default -- no selection
multsel_and_datef... 269 dclass = l['speaker']
multsel_and_datef... 270
multsel_and_datef... 271 # If selection is given:
multsel_and_datef... 272 if select != []:
multsel_and_datef... 273 ss, se = select
multsel_and_datef... 274 if ss <= l['idx'] <= se:
multsel_and_datef... 275 dclass = "highlight"
multsel_and_datef... 276
logotron_genesis.kv 277 # HTMLize the given line :
multsel_and_datef... 278 s = ("<div id='{0}' class='{6}{5}'>"
logotron_genesis.kv 279 "<a class='nick' title='{2}'"
logotron_genesis.kv 280 " href=\"{3}\">{1}</a>: {4}</div>").format(l['idx'],
logotron_genesis.kv 281 l['speaker'],
logotron_genesis.kv 282 l['t'],
logotron_genesis.kv 283 line_url(l),
logotron_genesis.kv 284 payload,
multsel_and_datef... 285 bot,
multsel_and_datef... 286 dclass)
logotron_genesis.kv 287 return s
logotron_genesis.kv 288
logotron_genesis.kv 289 # Make above callable from inside htm templater:
logotron_genesis.kv 290 app.jinja_env.globals.update(format_logline=format_logline)
logotron_genesis.kv 291
logotron_genesis.kv 292
logotron_genesis.kv 293 # Generate navbar for the given date:
logotron_genesis.kv 294 def generate_navbar(date, tail, chan):
logotron_genesis.kv 295 cur_day = datetime.strptime(date, Date_Short_Format)
logotron_genesis.kv 296 prev_day = cur_day - timedelta(days=1)
logotron_genesis.kv 297 prev_day_txt = prev_day.strftime(Date_Short_Format)
logotron_genesis.kv 298
logotron_genesis.kv 299 s = "<a href='{0}log/{1}/{2}'>← {2}</a>".format(
logotron_genesis.kv 300 get_base(),
logotron_genesis.kv 301 chan,
logotron_genesis.kv 302 prev_day_txt)
logotron_genesis.kv 303
logotron_genesis.kv 304 if not tail:
logotron_genesis.kv 305 next_day = cur_day + timedelta(days=1)
logotron_genesis.kv 306 next_day_txt = next_day.strftime(Date_Short_Format)
logotron_genesis.kv 307 s = s + " | <a href='{0}log/{1}/{2}'>{2} →</a>".format(
logotron_genesis.kv 308 get_base(),
logotron_genesis.kv 309 chan,
logotron_genesis.kv 310 next_day_txt)
logotron_genesis.kv 311
logotron_genesis.kv 312 return s
logotron_genesis.kv 313
logotron_genesis.kv 314 # Make above callable from inside htm templater:
logotron_genesis.kv 315 app.jinja_env.globals.update(generate_navbar=generate_navbar)
logotron_genesis.kv 316
logotron_genesis.kv 317
logotron_genesis.kv 318 @app.route('/log/<chan>/<date>')
logotron_genesis.kv 319 @app.route('/log/<chan>', defaults={'date': None})
logotron_genesis.kv 320 @app.route('/log/', defaults={'chan': Default_Chan, 'date': None})
logotron_genesis.kv 321 @app.route('/log', defaults={'chan': Default_Chan, 'date': None})
logotron_genesis.kv 322 def log(chan, date):
logotron_genesis.kv 323 # Handle rubbish chan:
logotron_genesis.kv 324 if chan not in Channels:
logotron_genesis.kv 325 return redirect(url_for('log'))
logotron_genesis.kv 326
multsel_and_datef... 327 # Get possible selection start and end
multsel_and_datef... 328 sel_start = request.args.get('ss', default = 0, type = int)
multsel_and_datef... 329 sel_end = request.args.get('se', default = 0, type = int)
multsel_and_datef... 330
logotron_genesis.kv 331 # Get current time
logotron_genesis.kv 332 now = datetime.now()
logotron_genesis.kv 333
logotron_genesis.kv 334 # Whether we are viewing 'current' tail
logotron_genesis.kv 335 tail = False
logotron_genesis.kv 336
logotron_genesis.kv 337 # If viewing 'current' log:
logotron_genesis.kv 338 if date == None:
logotron_genesis.kv 339 date = now.strftime(Date_Short_Format)
logotron_genesis.kv 340 tail = True
logotron_genesis.kv 341
logotron_genesis.kv 342 # Parse given date, and redirect to default log if rubbish:
logotron_genesis.kv 343 try:
logotron_genesis.kv 344 day_start = datetime.strptime(date, Date_Short_Format)
logotron_genesis.kv 345 except Exception, e:
logotron_genesis.kv 346 return redirect(url_for('log'))
logotron_genesis.kv 347
logotron_genesis.kv 348 # Determine the end of the interval being shown
logotron_genesis.kv 349 day_end = day_start + timedelta(days=1)
logotron_genesis.kv 350
multsel_and_datef... 351 # Enable 'tail' is day_end is after end of current day
multsel_and_datef... 352 if day_end.day > now.day:
multsel_and_datef... 353 tail = True
multsel_and_datef... 354
logotron_genesis.kv 355 # Get the loglines from DB
logotron_genesis.kv 356 lines = query_db(
logotron_genesis.kv 357 '''select * from loglines where chan=%s
logotron_genesis.kv 358 and t between %s and %s order by idx asc;''',
logotron_genesis.kv 359 [chan, day_start, day_end], one=False)
logotron_genesis.kv 360
logotron_genesis.kv 361 # Return the HTMLized text
logotron_genesis.kv 362 return render_template('log.html',
logotron_genesis.kv 363 chan = chan,
logotron_genesis.kv 364 loglines = lines,
multsel_and_datef... 365 sel = (sel_start, sel_end),
logotron_genesis.kv 366 date = date,
logotron_genesis.kv 367 tail = tail)
logotron_genesis.kv 368
logotron_genesis.kv 369
raw_line_export.kv 370 @app.route('/log-raw/<chan>')
raw_line_export.kv 371 def rawlog(chan):
raw_line_export.kv 372 res = ""
raw_line_export.kv 373
raw_line_export.kv 374 # Handle rubbish chan:
raw_line_export.kv 375 if chan not in Channels:
raw_line_export.kv 376 return Response("EGGOG: No such Channel!", mimetype='text/plain')
raw_line_export.kv 377
raw_line_export.kv 378 # Get start and end indices:
raw_line_export.kv 379 idx_start = request.args.get('istart', default = 0, type = int)
raw_line_export.kv 380 idx_end = request.args.get('iend', default = 0, type = int)
raw_line_export.kv 381
raw_line_export.kv 382 # Malformed bounds?
raw_line_export.kv 383 if idx_start > idx_end:
raw_line_export.kv 384 return Response("EGGOG: Start must precede End!",
raw_line_export.kv 385 mimetype='text/plain')
raw_line_export.kv 386
raw_line_export.kv 387 # Demanded too many in one burst ?
raw_line_export.kv 388 if (idx_end - idx_start) > Max_Raw_Ln :
raw_line_export.kv 389 return Response("EGGOG: May request Max. of %s Lines !" % Max_Raw_Ln,
raw_line_export.kv 390 mimetype='text/plain')
raw_line_export.kv 391
raw_line_export.kv 392 # Get the loglines from DB
raw_line_export.kv 393 lines = query_db(
raw_line_export.kv 394 '''select * from loglines where chan=%s
raw_line_export.kv 395 and idx between %s and %s order by idx asc;''',
raw_line_export.kv 396 [chan, idx_start, idx_end], one=False)
raw_line_export.kv 397
raw_line_export.kv 398 # Retrieve raw lines in classical Phf format:
raw_line_export.kv 399 for l in lines:
raw_line_export.kv 400 action = ""
raw_line_export.kv 401 if l['self']:
raw_line_export.kv 402 action = "*;"
raw_line_export.kv 403 res += "%s;%s;%s%s;%s\n" % (l['idx'],
raw_line_export.kv 404 l['t'].strftime('%s'),
raw_line_export.kv 405 action,
raw_line_export.kv 406 l['speaker'],
raw_line_export.kv 407 l['payload'])
raw_line_export.kv 408
raw_line_export.kv 409 # Return plain text:
raw_line_export.kv 410 return Response(res, mimetype='text/plain')
raw_line_export.kv 411
logotron_genesis.kv 412
logotron_genesis.kv 413 Name_Chars = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ_-"
logotron_genesis.kv 414
logotron_genesis.kv 415 def sanitize_speaker(s):
logotron_genesis.kv 416 return "".join([ch for ch in s if ch in Name_Chars])
logotron_genesis.kv 417
logotron_genesis.kv 418
logotron_genesis.kv 419 def re_escape(s):
logotron_genesis.kv 420 return re.sub(r"[(){}\[\].*?|^$\\+-]", r"\\\g<0>", s)
logotron_genesis.kv 421
logotron_genesis.kv 422 # Search knob. Supports 'chan' parameter.
logotron_genesis.kv 423 @app.route('/log-search')
logotron_genesis.kv 424 def logsearch():
logotron_genesis.kv 425 # The query params:
logotron_genesis.kv 426 chan = request.args.get('chan', default = Default_Chan, type = str)
logotron_genesis.kv 427 query = request.args.get('q', default = '', type = str)
logotron_genesis.kv 428 # page_num = request.args.get('page', default = 0, type = int)
logotron_genesis.kv 429
logotron_genesis.kv 430 # Handle rubbish chan:
logotron_genesis.kv 431 if chan not in Channels:
logotron_genesis.kv 432 return redirect(url_for('log'))
logotron_genesis.kv 433
logotron_genesis.kv 434 nres = 0
logotron_genesis.kv 435 searchres = []
logotron_genesis.kv 436 tokens_orig = []
logotron_genesis.kv 437 search_head = "Query is too short!"
logotron_genesis.kv 438 # Forbid query that is too short:
logotron_genesis.kv 439 if len(query) >= Min_Query_Length:
logotron_genesis.kv 440 # Get the search tokens to use:
logotron_genesis.kv 441 tokens = query.split()
logotron_genesis.kv 442 tokens_standard = []
logotron_genesis.kv 443 from_users = []
logotron_genesis.kv 444
logotron_genesis.kv 445 # separate out "from:foo" tokens and ordinary:
logotron_genesis.kv 446 for t in tokens:
logotron_genesis.kv 447 if t.startswith("from:") or t.startswith("f:"):
logotron_genesis.kv 448 from_users.append(t.split(':')[1]) # Record user for 'from' query
logotron_genesis.kv 449 else:
logotron_genesis.kv 450 tokens_standard.append(t)
logotron_genesis.kv 451
logotron_genesis.kv 452 from_users = ['%' + sanitize_speaker(t) + '%' for t in from_users]
logotron_genesis.kv 453 tokens_orig = [re_escape(t) for t in tokens_standard]
logotron_genesis.kv 454 tokens_formed = ['%' + t + '%' for t in tokens_orig]
logotron_genesis.kv 455
logotron_genesis.kv 456 # Query is usable; perform the search on DB and get the finds
logotron_genesis.kv 457 if from_users == []:
logotron_genesis.kv 458 searchres = query_db(
logotron_genesis.kv 459 '''select * from loglines where chan=%s
logotron_genesis.kv 460 and payload ilike all(%s) order by idx desc limit %s;''',
logotron_genesis.kv 461 [chan,
logotron_genesis.kv 462 tokens_formed,
logotron_genesis.kv 463 Max_Search_Results], one=False)
logotron_genesis.kv 464 else:
logotron_genesis.kv 465 searchres = query_db(
logotron_genesis.kv 466 '''select * from loglines where chan=%s
logotron_genesis.kv 467 and speaker ilike any(%s)
logotron_genesis.kv 468 and payload ilike all(%s) order by idx desc limit %s;''',
logotron_genesis.kv 469 [chan,
logotron_genesis.kv 470 from_users,
logotron_genesis.kv 471 tokens_formed,
logotron_genesis.kv 472 Max_Search_Results], one=False)
logotron_genesis.kv 473
logotron_genesis.kv 474
logotron_genesis.kv 475 # Number of entries found
logotron_genesis.kv 476 nres = len(searchres)
logotron_genesis.kv 477 search_head = "<b>{0}</b> entries found in {1} for <b>'{2}'</b> :".format(
logotron_genesis.kv 478 nres, chan, html_escape(query))
logotron_genesis.kv 479
logotron_genesis.kv 480 # No paging support just yet:
logotron_genesis.kv 481 return render_template('searchres.html',
logotron_genesis.kv 482 query = query,
logotron_genesis.kv 483 nres = nres,
logotron_genesis.kv 484 chan = chan,
logotron_genesis.kv 485 search_head = search_head,
logotron_genesis.kv 486 tokens = tokens_orig,
logotron_genesis.kv 487 loglines = searchres)
logotron_genesis.kv 488
logotron_genesis.kv 489
logotron_genesis.kv 490 # Comment this out if you don't have one
logotron_genesis.kv 491 @app.route('/favicon.ico')
logotron_genesis.kv 492 def favicon():
logotron_genesis.kv 493 return redirect(url_for('static', filename='favicon.ico'))
logotron_genesis.kv 494
logotron_genesis.kv 495
logotron_genesis.kv 496 ## App Mode
logotron_genesis.kv 497 if __name__ == '__main__':
logotron_genesis.kv 498 app.run(threaded=True, port=WWW_Port)