diff options
Diffstat (limited to 'views.py')
| -rw-r--r-- | views.py | 253 |
1 files changed, 253 insertions, 0 deletions
diff --git a/views.py b/views.py new file mode 100644 index 0000000..070754e --- /dev/null +++ b/views.py @@ -0,0 +1,253 @@ +# This file is part of GNU Dico. +# Copyright (C) 2008-2010, 2012 Wojciech Polak +# +# GNU Dico is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 3, or (at your option) +# any later version. +# +# GNU Dico is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with GNU Dico. If not, see <http://www.gnu.org/licenses/>. + +import re +import hashlib +import socket +from django.conf import settings +from django.core import urlresolvers +from django.core.cache import cache +from django.shortcuts import render_to_response +from django.utils.translation import ugettext as _ + +import dicoclient +from wit import wiki2html + +known_items = [ 'about', 'license', 'download' ] + +def index (request, item=None): + page = { + 'robots': 'index', + } + selects = {} + mtc = {} + result = {} + markup_style = 'default' + port = 2628 + databases = [] + strategies = [] + + menu_class = {} + + if item and not item in known_items: + return render_to_response ('404.html', {}) + + sid = request.COOKIES.get ('dicoweb_sid', '') + request.session.set_expiry (0) + + accept_lang = request.META.get ('HTTP_ACCEPT_LANGUAGE', '').split (',') + for i, lang in enumerate (accept_lang): + accept_lang[i] = lang.split (';')[0] + + if 'server' in request.session: + server = request.session['server'] + else: + server = settings.DICT_SERVERS[0] + server = request.GET.get ('server', server) + if not server in settings.DICT_SERVERS: + server = settings.DICT_SERVERS[0] + request.session['server'] = server; + + if len (settings.DICT_SERVERS) > 1: + selects['sv'] = HtmlOptions (settings.DICT_SERVERS, server) + + key = hashlib.md5 ('%s/%s' % (sid, server.encode ('ascii', + 'backslashreplace'))) + sid = key.hexdigest () + + type = 'search' + if 'define' in request.GET: + type = 'define' + else: + type = 'search' + + database = request.GET.get ('db', '*') + strategy = request.GET.get ('strategy', '.') + + if (server in settings.AUTH and 'user' in settings.AUTH[server]): + cred = settings.AUTH[server] + srvkey = cred['user'] + '@' + server + else: + cred = None + srvkey = server + + key_databases = str ('dicoweb/databases/' + srvkey) + key_strategies = str ('dicoweb/strategies/' + srvkey) + databases = cache.get (key_databases) + strategies = cache.get (key_strategies) + + if server.find (':') != -1: + s = server.split (':', 1) + server = s[0] + port = int (s[1]) + + if not databases or not strategies: + dc = dicoclient.DicoClient () + try: + dc.open (server, port, cred) + dc.timeout = settings.DICT_TIMEOUT + databases = dc.show_databases ()['databases'] + strategies = dc.show_strategies ()['strategies'] + dc.close () + except (socket.timeout, socket.error, dicoclient.DicoNotConnectedError): + return render_to_response ('index.html', {'selects': selects}) + + cache.set (key_databases, databases, timeout=86400) + cache.set (key_strategies, strategies, timeout=86400) + + for s in strategies: + s[1] = _(s[1]) + strategies.insert (0, ['.', _('Default')]) + + selects['st'] = HtmlOptions (strategies, strategy) + + if (len(databases) > 1): + selects['db'] = HtmlOptions (databases, database) + databases.insert (0, ['!', _('First match')]) + databases.insert (0, ['*', _('All')]) + + q = request.GET.get ('q', '') + + if 'q' in request.GET and q != '': + langkey = '*'; + if database == '*': langkey = ','.join (accept_lang) + + key = hashlib.md5 ('%s:%d/%s/%s/%s/%s/%s' % + (srvkey, port, langkey, type, database, strategy, + q.encode ('ascii', 'backslashreplace'))) + key = key.hexdigest () + result = cache.get ('dicoweb/' + key) + + if not result: + try: + dc = dicoclient.DicoClient () + dc.timeout = settings.DICT_TIMEOUT + dc.open (server, port, cred) + dc.option ('MIME') + + if database == '*' and 'lang' in dc.server_capas: + dc.option ('LANG', ': ' + ' '.join (accept_lang)) + if 'markup-wiki' in dc.server_capas: + if dc.option ('MARKUP', 'wiki'): + markup_style = 'wiki' + + if database == 'dbinfo': + result = dc.show_info (q) + elif type == 'define': + result = dc.define (database, q) + if 'error' in result and result['error'] == '552': + result = dc.match (database, strategy, q) + type = 'search' + result['type'] = 'define' + else: + result = dc.match (database, strategy, q) + result['type'] = 'search' + dc.close () + + result['markup_style'] = markup_style + cache.set ('dicoweb/' + key, result, timeout=3600) + + except (socket.timeout, socket.error, + dicoclient.DicoNotConnectedError): + return render_to_response ('index.html', + {'selects': selects}) + + # get last match results + if sid and type == 'search': + cache.set ('dicoweb/%s/last_match' % sid, key, timeout=3600) + else: + key = cache.get ('dicoweb/%s/last_match' % sid) + if key != None: + mtc = cache.get ('dicoweb/' + key) + + mtc['dbnames'] = {} + if 'matches' in mtc: + for m in mtc['matches']: + for d in databases: + if d[0] == m: + mtc['dbnames'][m] = d[1] + break + + if database == 'dbinfo': q = '' + if q != '': + page['title'] = q + ' - ' + page['robots'] = 'noindex,nofollow' + else: + if not item: + item = 'about' + menu_class[item] = 'class="active"' + item += ".html" + + if 'definitions' in result: + rx1 = re.compile ('{+(.*?)}+', re.DOTALL) + for df in result['definitions']: + if df.has_key ('content-type') \ + and df['content-type'].startswith ('text/x-wiki'): + lang = df['x-wiki-language'] \ + if df.has_key ('x-wiki-language') else 'en' + wikiparser = wiki2html.HtmlWiktionaryMarkup (text=df['desc'], + html_base='?q=', + lang=lang) + wikiparser.parse () + df['desc'] = str (wikiparser) + df['format_html'] = True + else: + df['desc'] = re.sub ('_(.*?)_', '<b>\\1</b>', df['desc']) + df['desc'] = re.sub (rx1, __subs1, df['desc']) + + return render_to_response ('index.html', {'page': page, + 'item': item, + 'menu_class': menu_class, + 'q': q, + 'mtc': mtc, + 'result': result, + 'selects': selects,}) + +def opensearch (request): + url_query = request.build_absolute_uri (urlresolvers.reverse ('index')) + url_media = request.build_absolute_uri (settings.MEDIA_URL) + return render_to_response ('opensearch.xml', {'url_query': url_query, + 'url_media': url_media}, + mimetype='application/xml') + +def __subs1 (match): + s = re.sub (r' +', ' ', match.group (1)) + # FIXME: added "define=Define", which is in my opinion right. + return '<a href="?q=%s&define=Define" title="Search for %s">%s</a>' \ + % (s.replace ('\n', ''), s.replace ('\n', ''), s) + +class HtmlOptions: + def __init__ (self, lst=[], value=''): + self.lst = lst + self.value = value + + def html (self): + buf = [] + for opt in self.lst: + if len (opt) == 2: + if not opt[1]: + opt[1] = opt[0] + if opt[0] == self.value: + buf.append ('<option value="%s" selected="selected">%s</option>' % (opt[0], opt[1])) + else: + buf.append ('<option value="%s">%s</option>' % (opt[0], + opt[1])) + else: + if opt == self.value: + buf.append ('<option value="%s" selected="selected">%s</option>' % (opt, opt)) + else: + buf.append ('<option value="%s">%s</option>' % (opt, opt)) + return '\n'.join (buf) |