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:
uniturds_etc.kv 246 boxlinks_re = re.compile(
uniturds_etc.kv 247 '\[\s*<a href="(http[^ \[\]]+)"[^>]*>[^ <]+</a>\s*\]\[([^\[\]]+)\]')
uniturds_etc.kv 248
logotron_genesis.kv 249 stdlinks_re = re.compile('(http[^ \[\]]+)')
logotron_genesis.kv 250
logotron_genesis.kv 251
logotron_genesis.kv 252 ## Format given log line for display
multsel_and_datef... 253 def format_logline(l, highlights = [], select=[]):
logotron_genesis.kv 254 payload = html_escape(l['payload'])
logotron_genesis.kv 255
logotron_genesis.kv 256 # Format ordinary links:
uniturds_etc.kv 257 payload = re.sub(stdlinks_re,
uniturds_etc.kv 258 r'<a href="\1" target=\'_blank\'>\1</a>', payload)
logotron_genesis.kv 259
logotron_genesis.kv 260 # Now also format [link][text] links :
uniturds_etc.kv 261 payload = re.sub(boxlinks_re,
uniturds_etc.kv 262 r'<a href="\1" target=\'_blank\'>\2</a>', payload)
logotron_genesis.kv 263
logotron_genesis.kv 264 # If this is a search result, illuminate the matched strings:
logotron_genesis.kv 265 if highlights != []:
logotron_genesis.kv 266 payload = highlight_text(highlights, payload)
logotron_genesis.kv 267
logotron_genesis.kv 268 bot = ""
logotron_genesis.kv 269 if l['speaker'] in Bots:
logotron_genesis.kv 270 bot = " bot"
logotron_genesis.kv 271
multsel_and_datef... 272 # default -- no selection
multsel_and_datef... 273 dclass = l['speaker']
multsel_and_datef... 274
multsel_and_datef... 275 # If selection is given:
multsel_and_datef... 276 if select != []:
multsel_and_datef... 277 ss, se = select
multsel_and_datef... 278 if ss <= l['idx'] <= se:
multsel_and_datef... 279 dclass = "highlight"
multsel_and_datef... 280
logotron_genesis.kv 281 # HTMLize the given line :
multsel_and_datef... 282 s = ("<div id='{0}' class='{6}{5}'>"
logotron_genesis.kv 283 "<a class='nick' title='{2}'"
logotron_genesis.kv 284 " href=\"{3}\">{1}</a>: {4}</div>").format(l['idx'],
logotron_genesis.kv 285 l['speaker'],
logotron_genesis.kv 286 l['t'],
logotron_genesis.kv 287 line_url(l),
logotron_genesis.kv 288 payload,
multsel_and_datef... 289 bot,
multsel_and_datef... 290 dclass)
logotron_genesis.kv 291 return s
logotron_genesis.kv 292
logotron_genesis.kv 293 # Make above callable from inside htm templater:
logotron_genesis.kv 294 app.jinja_env.globals.update(format_logline=format_logline)
logotron_genesis.kv 295
logotron_genesis.kv 296
logotron_genesis.kv 297 # Generate navbar for the given date:
logotron_genesis.kv 298 def generate_navbar(date, tail, chan):
logotron_genesis.kv 299 cur_day = datetime.strptime(date, Date_Short_Format)
logotron_genesis.kv 300 prev_day = cur_day - timedelta(days=1)
logotron_genesis.kv 301 prev_day_txt = prev_day.strftime(Date_Short_Format)
logotron_genesis.kv 302
logotron_genesis.kv 303 s = "<a href='{0}log/{1}/{2}'>← {2}</a>".format(
logotron_genesis.kv 304 get_base(),
logotron_genesis.kv 305 chan,
logotron_genesis.kv 306 prev_day_txt)
logotron_genesis.kv 307
logotron_genesis.kv 308 if not tail:
logotron_genesis.kv 309 next_day = cur_day + timedelta(days=1)
logotron_genesis.kv 310 next_day_txt = next_day.strftime(Date_Short_Format)
logotron_genesis.kv 311 s = s + " | <a href='{0}log/{1}/{2}'>{2} →</a>".format(
logotron_genesis.kv 312 get_base(),
logotron_genesis.kv 313 chan,
logotron_genesis.kv 314 next_day_txt)
logotron_genesis.kv 315
logotron_genesis.kv 316 return s
logotron_genesis.kv 317
logotron_genesis.kv 318 # Make above callable from inside htm templater:
logotron_genesis.kv 319 app.jinja_env.globals.update(generate_navbar=generate_navbar)
logotron_genesis.kv 320
logotron_genesis.kv 321
uniturds_etc.kv 322 @app.route('/rnd/<chan>')
uniturds_etc.kv 323 def rnd(chan):
uniturds_etc.kv 324 # Handle rubbish chan:
uniturds_etc.kv 325 if chan not in Channels:
uniturds_etc.kv 326 return redirect(url_for('log'))
uniturds_etc.kv 327
uniturds_etc.kv 328 rnd_line = query_db(
uniturds_etc.kv 329 '''select * from loglines where chan=%s
uniturds_etc.kv 330 order by random() limit 1 ;''',
uniturds_etc.kv 331 [chan], one=True)
uniturds_etc.kv 332
uniturds_etc.kv 333 return redirect(line_url(rnd_line))
uniturds_etc.kv 334
uniturds_etc.kv 335
logotron_genesis.kv 336 @app.route('/log/<chan>/<date>')
logotron_genesis.kv 337 @app.route('/log/<chan>', defaults={'date': None})
logotron_genesis.kv 338 @app.route('/log/', defaults={'chan': Default_Chan, 'date': None})
logotron_genesis.kv 339 @app.route('/log', defaults={'chan': Default_Chan, 'date': None})
logotron_genesis.kv 340 def log(chan, date):
logotron_genesis.kv 341 # Handle rubbish chan:
logotron_genesis.kv 342 if chan not in Channels:
logotron_genesis.kv 343 return redirect(url_for('log'))
logotron_genesis.kv 344
multsel_and_datef... 345 # Get possible selection start and end
multsel_and_datef... 346 sel_start = request.args.get('ss', default = 0, type = int)
multsel_and_datef... 347 sel_end = request.args.get('se', default = 0, type = int)
uniturds_etc.kv 348
uniturds_etc.kv 349 # Get possible 'reverse gear'
uniturds_etc.kv 350 rev = request.args.get('rev', default = 0, type = int)
multsel_and_datef... 351
logotron_genesis.kv 352 # Get current time
logotron_genesis.kv 353 now = datetime.now()
logotron_genesis.kv 354
logotron_genesis.kv 355 # Whether we are viewing 'current' tail
logotron_genesis.kv 356 tail = False
logotron_genesis.kv 357
logotron_genesis.kv 358 # If viewing 'current' log:
logotron_genesis.kv 359 if date == None:
logotron_genesis.kv 360 date = now.strftime(Date_Short_Format)
logotron_genesis.kv 361 tail = True
logotron_genesis.kv 362
logotron_genesis.kv 363 # Parse given date, and redirect to default log if rubbish:
logotron_genesis.kv 364 try:
logotron_genesis.kv 365 day_start = datetime.strptime(date, Date_Short_Format)
logotron_genesis.kv 366 except Exception, e:
logotron_genesis.kv 367 return redirect(url_for('log'))
logotron_genesis.kv 368
logotron_genesis.kv 369 # Determine the end of the interval being shown
logotron_genesis.kv 370 day_end = day_start + timedelta(days=1)
logotron_genesis.kv 371
multsel_and_datef... 372 # Enable 'tail' is day_end is after end of current day
rle_errata.kv 373 if day_end > now:
multsel_and_datef... 374 tail = True
multsel_and_datef... 375
logotron_genesis.kv 376 # Get the loglines from DB
logotron_genesis.kv 377 lines = query_db(
logotron_genesis.kv 378 '''select * from loglines where chan=%s
logotron_genesis.kv 379 and t between %s and %s order by idx asc;''',
logotron_genesis.kv 380 [chan, day_start, day_end], one=False)
logotron_genesis.kv 381
uniturds_etc.kv 382 # Optional 'reverse gear' knob:
uniturds_etc.kv 383 if rev == 1:
uniturds_etc.kv 384 lines.reverse()
uniturds_etc.kv 385
logotron_genesis.kv 386 # Return the HTMLized text
logotron_genesis.kv 387 return render_template('log.html',
logotron_genesis.kv 388 chan = chan,
logotron_genesis.kv 389 loglines = lines,
multsel_and_datef... 390 sel = (sel_start, sel_end),
logotron_genesis.kv 391 date = date,
uniturds_etc.kv 392 tail = tail,
uniturds_etc.kv 393 rev = not rev)
logotron_genesis.kv 394
logotron_genesis.kv 395
raw_line_export.kv 396 @app.route('/log-raw/<chan>')
raw_line_export.kv 397 def rawlog(chan):
raw_line_export.kv 398 res = ""
raw_line_export.kv 399
raw_line_export.kv 400 # Handle rubbish chan:
raw_line_export.kv 401 if chan not in Channels:
raw_line_export.kv 402 return Response("EGGOG: No such Channel!", mimetype='text/plain')
raw_line_export.kv 403
raw_line_export.kv 404 # Get start and end indices:
raw_line_export.kv 405 idx_start = request.args.get('istart', default = 0, type = int)
raw_line_export.kv 406 idx_end = request.args.get('iend', default = 0, type = int)
raw_line_export.kv 407
raw_line_export.kv 408 # Malformed bounds?
raw_line_export.kv 409 if idx_start > idx_end:
raw_line_export.kv 410 return Response("EGGOG: Start must precede End!",
raw_line_export.kv 411 mimetype='text/plain')
raw_line_export.kv 412
raw_line_export.kv 413 # Demanded too many in one burst ?
raw_line_export.kv 414 if (idx_end - idx_start) > Max_Raw_Ln :
raw_line_export.kv 415 return Response("EGGOG: May request Max. of %s Lines !" % Max_Raw_Ln,
raw_line_export.kv 416 mimetype='text/plain')
raw_line_export.kv 417
raw_line_export.kv 418 # Get the loglines from DB
raw_line_export.kv 419 lines = query_db(
raw_line_export.kv 420 '''select * from loglines where chan=%s
raw_line_export.kv 421 and idx between %s and %s order by idx asc;''',
raw_line_export.kv 422 [chan, idx_start, idx_end], one=False)
raw_line_export.kv 423
raw_line_export.kv 424 # Retrieve raw lines in classical Phf format:
raw_line_export.kv 425 for l in lines:
raw_line_export.kv 426 action = ""
raw_line_export.kv 427 if l['self']:
raw_line_export.kv 428 action = "*;"
raw_line_export.kv 429 res += "%s;%s;%s%s;%s\n" % (l['idx'],
raw_line_export.kv 430 l['t'].strftime('%s'),
raw_line_export.kv 431 action,
raw_line_export.kv 432 l['speaker'],
raw_line_export.kv 433 l['payload'])
raw_line_export.kv 434
raw_line_export.kv 435 # Return plain text:
raw_line_export.kv 436 return Response(res, mimetype='text/plain')
raw_line_export.kv 437
logotron_genesis.kv 438
logotron_genesis.kv 439 Name_Chars = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ_-"
logotron_genesis.kv 440
logotron_genesis.kv 441 def sanitize_speaker(s):
logotron_genesis.kv 442 return "".join([ch for ch in s if ch in Name_Chars])
logotron_genesis.kv 443
logotron_genesis.kv 444
logotron_genesis.kv 445 def re_escape(s):
logotron_genesis.kv 446 return re.sub(r"[(){}\[\].*?|^$\\+-]", r"\\\g<0>", s)
logotron_genesis.kv 447
logotron_genesis.kv 448 # Search knob. Supports 'chan' parameter.
logotron_genesis.kv 449 @app.route('/log-search')
logotron_genesis.kv 450 def logsearch():
logotron_genesis.kv 451 # The query params:
logotron_genesis.kv 452 chan = request.args.get('chan', default = Default_Chan, type = str)
logotron_genesis.kv 453 query = request.args.get('q', default = '', type = str)
logotron_genesis.kv 454 # page_num = request.args.get('page', default = 0, type = int)
logotron_genesis.kv 455
logotron_genesis.kv 456 # Handle rubbish chan:
logotron_genesis.kv 457 if chan not in Channels:
logotron_genesis.kv 458 return redirect(url_for('log'))
logotron_genesis.kv 459
logotron_genesis.kv 460 nres = 0
logotron_genesis.kv 461 searchres = []
logotron_genesis.kv 462 tokens_orig = []
logotron_genesis.kv 463 search_head = "Query is too short!"
logotron_genesis.kv 464 # Forbid query that is too short:
logotron_genesis.kv 465 if len(query) >= Min_Query_Length:
logotron_genesis.kv 466 # Get the search tokens to use:
logotron_genesis.kv 467 tokens = query.split()
logotron_genesis.kv 468 tokens_standard = []
logotron_genesis.kv 469 from_users = []
logotron_genesis.kv 470
logotron_genesis.kv 471 # separate out "from:foo" tokens and ordinary:
logotron_genesis.kv 472 for t in tokens:
logotron_genesis.kv 473 if t.startswith("from:") or t.startswith("f:"):
logotron_genesis.kv 474 from_users.append(t.split(':')[1]) # Record user for 'from' query
logotron_genesis.kv 475 else:
logotron_genesis.kv 476 tokens_standard.append(t)
logotron_genesis.kv 477
logotron_genesis.kv 478 from_users = ['%' + sanitize_speaker(t) + '%' for t in from_users]
logotron_genesis.kv 479 tokens_orig = [re_escape(t) for t in tokens_standard]
logotron_genesis.kv 480 tokens_formed = ['%' + t + '%' for t in tokens_orig]
logotron_genesis.kv 481
logotron_genesis.kv 482 # Query is usable; perform the search on DB and get the finds
logotron_genesis.kv 483 if from_users == []:
logotron_genesis.kv 484 searchres = query_db(
logotron_genesis.kv 485 '''select * from loglines where chan=%s
logotron_genesis.kv 486 and payload ilike all(%s) order by idx desc limit %s;''',
logotron_genesis.kv 487 [chan,
logotron_genesis.kv 488 tokens_formed,
logotron_genesis.kv 489 Max_Search_Results], one=False)
logotron_genesis.kv 490 else:
logotron_genesis.kv 491 searchres = query_db(
logotron_genesis.kv 492 '''select * from loglines where chan=%s
logotron_genesis.kv 493 and speaker ilike any(%s)
logotron_genesis.kv 494 and payload ilike all(%s) order by idx desc limit %s;''',
logotron_genesis.kv 495 [chan,
logotron_genesis.kv 496 from_users,
logotron_genesis.kv 497 tokens_formed,
logotron_genesis.kv 498 Max_Search_Results], one=False)
logotron_genesis.kv 499
logotron_genesis.kv 500
logotron_genesis.kv 501 # Number of entries found
logotron_genesis.kv 502 nres = len(searchres)
logotron_genesis.kv 503 search_head = "<b>{0}</b> entries found in {1} for <b>'{2}'</b> :".format(
logotron_genesis.kv 504 nres, chan, html_escape(query))
logotron_genesis.kv 505
logotron_genesis.kv 506 # No paging support just yet:
logotron_genesis.kv 507 return render_template('searchres.html',
logotron_genesis.kv 508 query = query,
logotron_genesis.kv 509 nres = nres,
logotron_genesis.kv 510 chan = chan,
logotron_genesis.kv 511 search_head = search_head,
logotron_genesis.kv 512 tokens = tokens_orig,
logotron_genesis.kv 513 loglines = searchres)
logotron_genesis.kv 514
logotron_genesis.kv 515
logotron_genesis.kv 516 # Comment this out if you don't have one
logotron_genesis.kv 517 @app.route('/favicon.ico')
logotron_genesis.kv 518 def favicon():
logotron_genesis.kv 519 return redirect(url_for('static', filename='favicon.ico'))
logotron_genesis.kv 520
logotron_genesis.kv 521
logotron_genesis.kv 522 ## App Mode
logotron_genesis.kv 523 if __name__ == '__main__':
logotron_genesis.kv 524 app.run(threaded=True, port=WWW_Port)