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