tko/save_query.cgi

   1 #!/usr/bin/python
   2
   3 import os, cgi, cgitb, time, urllib
   4 import db, unique_cookie
   5
   6 ## setting script globals
   7 form = cgi.FieldStorage()
   8 if 'label' in form.keys():
   9     comment = form['label'].value
  10 else:
  11     comment = ''
  12 dict_url = {}
  13 for key in form.keys():
  14     dict_url[key] = form[key].value
  15
  16 tm = time.asctime()
  17 uid = unique_cookie.unique_id('tko_history')
  18 HTTP_REFERER = os.environ.get('HTTP_REFERER')
  19 if HTTP_REFERER is None:
  20     ## fall back strategy for proxy connection
  21     ## substitute relative url
  22     HTTP_REFERER = 'compose_query.cgi?' + urllib.urlencode(dict_url)
  23
  24
  25 class QueryHistoryError(Exception):
  26     pass
  27
  28
  29 def log_query():
  30     db_obj = db.db()
  31     data_to_insert = {'uid':uid, 'time_created':tm,
  32               'user_comment':comment, 'url':HTTP_REFERER }
  33     try:
  34         db_obj.insert('tko_query_history', data_to_insert)
  35     except:
  36         raise QueryHistoryError("Could not save query")
  37
  38
  39 def delete_query(time_stamp):
  40     ## query is marked for delete by time stamp
  41     db_obj = db.db()
  42     data_to_delete = {'time_created':time_stamp}
  43     try:
  44         db_obj.delete('tko_query_history', data_to_delete)
  45     except Exception:
  46         raise QueryHistoryError("Could not delete query")
  47
  48
  49 def body():
  50     if not 'delete' in dict_url.keys():
  51         log_query()
  52         print '<b>%s</b><br><br>' % "Your query has been saved"
  53         print 'time: %s<br>' % tm
  54         print 'comments: %s<br><br>' % comment
  55     else:
  56         ## key 'delete' has arg value of time_stamp
  57         ## which identifies the query to be deleted
  58         time_stamp = dict_url['delete']
  59         delete_query(time_stamp)
  60         print '<b>%s</b><br><br>' % "Your query has been deleted"
  61
  62     print '<a href="query_history.cgi">View saved queries</a>&nbsp;&nbsp;'
  63     print '<br><br>'
  64     if not 'delete' in dict_url.keys():
  65         print '<a href="%s">Back to Autotest</a><br>' % HTTP_REFERER
  66     else:
  67         print '<a href="compose_query.cgi">Autotest Results</a><br>'
  68
  69
  70 def main():
  71     print "Content-type: text/html\n"
  72     print '<html><head><title>'
  73     print '</title></head>'
  74     print '<body>'
  75     body()
  76     print '</body>'
  77     print '</html>'
  78
  79
  80 main()