Version 0.8: Python 3

[phdru.name/cgi-bin/blog-ru/search-tags.git] / tags.py

diff --git a/tags.py b/tags.py
index df64a1c286a7e645a84239c0d6571a7af46ee89d..92a37060cc32c81f3bddb7838795cc683e8548e4 100644 (file)
--- a/tags.py
+++ b/tags.py
@@ -1,36 +1,55 @@
 blog_filename = "blog_dict.pickle"
 
 try:
-   import cPickle as pickle
+    import cPickle as pickle
 except ImportError:
-   import pickle
+    import pickle
 
 try:
-   blog_file = open(blog_filename, "rb")
+    blog_file = open('../../../../phdru.name/ru/' + blog_filename, "rb")
 except IOError:
-   blog_dict = {}
+    blog_dict = {}
 else:
-   blog_dict = pickle.load(blog_file)
-   blog_file.close()
+    blog_dict = pickle.load(blog_file)
+    blog_file.close()
 
 
 # blog_dict is a mapping
 # (year, month, day) => [list of (file, title, lead, tags)]
 
-def tag_exists(tag):
-    for posts in blog_dict.itervalues():
-        for _file, _title, _lead, _tags in posts:
-            if tag in _tags:
-                return True
-    return False
+# Add lower-case tags
+_new_dict = {}
+for (year, month, day), posts in blog_dict.items():
+    _new_dict[year, month, day] = _posts = []
+    for _file, _title, _lead, _tags in posts:
+        tags_lower = [tag.lower().replace(' ', '_') for tag in _tags]
+        _posts.append((_file, _title, _lead, _tags, tags_lower))
+blog_dict = _new_dict
+
+
+def real_tag(tag):
+    ltag = tag.lower().replace(' ', '_')
+    for posts in blog_dict.values():
+        for _file, _title, _lead, _tags, _tags_lower in posts:
+            try:
+                ix = _tags_lower.index(ltag)
+            except ValueError:
+                continue
+            else:
+                return _tags[ix].replace(' ', '_')
+
 
 def _test_post(post, tree):
+    """Test if the list of tags in the post satisfies condition
+
+    Recursively evaluate the tree against the list of tags in the post.
+
+    """
     op = tree[0]
     if op == 'NAME':
         tag = tree[1]
         assert isinstance(tag, str)
-        _tags = post[3]
-        return tag in _tags
+        return tag.lower().replace(' ', '_') in post[4]
     elif op in ('AND', 'OR'):
         value1 = _test_post(post, tree[1])
         value2 = _test_post(post, tree[2])
@@ -40,16 +59,26 @@ def _test_post(post, tree):
             return value1 or value2
     elif op == 'NOT':
         return not _test_post(post, tree[1])
+    elif op == 'PARENS':
+        return _test_post(post, tree[1])
     else:
         raise ValueError("Cannot get there")
 
+
 def find_tags(tree):
+    """Test every blog post against parsed expression
+
+    Return all posts that passed the test.
+
+    """
     _posts = []
-    for (year, month, day), posts in blog_dict.iteritems():
+    for (year, month, day), posts in blog_dict.items():
         for post in posts:
             if _test_post(post, tree):
                 _posts.append((
-                    '/'.join((year, month, day, post[0].replace('.tmpl', '.html'))),
+                    year, month, day,
+                    '/'.join(
+                        (year, month, day, post[0][:-len("tmpl")] + "html")),
                     post[1]))
-    _posts.sort()
+    _posts.sort(reverse=True)
     return _posts