author | Tero Marttila <terom@paivola.fi> |
Sun, 14 Sep 2014 01:45:17 +0300 | |
changeset 201 | edcbbc94f90d |
parent 200 | 1757261f2b95 |
child 65 | 67d8600fa0a1 |
permissions | -rw-r--r-- |
197
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
1 |
from django.conf import settings |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
2 |
|
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
3 |
import codecs |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
4 |
import logging; log = logging.getLogger('qmsk_www_pages.pages') |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
5 |
import os, os.path |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
6 |
|
199 | 7 |
class Tree (object): |
200 | 8 |
INDEX = 'index' |
9 |
||
199 | 10 |
@classmethod |
11 |
def lookup (cls, parts): |
|
12 |
path = settings.QMSK_WWW_PAGES_DIR |
|
13 |
||
14 |
for part in parts: |
|
15 |
if part.startswith('.'): |
|
16 |
# evil |
|
17 |
return None |
|
18 |
||
19 |
if not part: |
|
20 |
continue |
|
21 |
||
22 |
path = os.path.join(path, part) |
|
23 |
||
24 |
if not os.path.exists(path): |
|
25 |
return None |
|
26 |
||
27 |
if not os.path.isdir(path): |
|
28 |
return None |
|
29 |
||
30 |
return cls(path, parts) |
|
31 |
||
32 |
def __init__ (self, path, parts): |
|
33 |
self.path = path |
|
34 |
self.parts = parts |
|
35 |
||
200 | 36 |
|
37 |
def url (self, tree=None, page=None): |
|
38 |
path = '/'.join(self.parts) |
|
39 |
||
40 |
if path: |
|
41 |
path += '/' |
|
42 |
||
43 |
if tree: |
|
44 |
path = tree + '/' |
|
45 |
||
46 |
if page: |
|
47 |
path += page |
|
48 |
||
49 |
return path |
|
50 |
||
201 | 51 |
def breadcrumb (self): |
52 |
path = [] |
|
53 |
||
54 |
yield '', 'Home' |
|
55 |
||
56 |
for part in self.parts: |
|
57 |
path.append(part) |
|
58 |
||
59 |
yield '/'.join(path) + '/', part |
|
60 |
||
200 | 61 |
def scan (self): |
62 |
""" |
|
63 |
Scan for files in tree. |
|
64 |
""" |
|
65 |
||
66 |
for filename in os.listdir(self.path): |
|
67 |
if filename.startswith('.'): |
|
68 |
continue |
|
69 |
||
70 |
if '.' in filename: |
|
71 |
file_name, file_type = filename.rsplit('.', 1) |
|
72 |
else: |
|
73 |
file_name = filename |
|
74 |
file_type = None |
|
75 |
||
76 |
if not file_name: |
|
77 |
continue |
|
78 |
||
79 |
path = os.path.join(self.path, filename) |
|
80 |
||
81 |
yield path, file_name, file_type |
|
82 |
||
83 |
def list (self): |
|
84 |
""" |
|
85 |
Lists all Trees and Pages for this Tree |
|
86 |
""" |
|
87 |
||
88 |
for path, file_name, file_type in self.scan(): |
|
89 |
# trees |
|
90 |
if os.path.isdir(path): |
|
91 |
yield self.url(tree=file_name), file_name |
|
92 |
||
93 |
if file_name == self.INDEX: |
|
94 |
continue |
|
95 |
||
96 |
# pages |
|
97 |
if not file_type: |
|
98 |
continue |
|
99 |
||
100 |
if file_type not in TYPES: |
|
101 |
continue |
|
102 |
||
103 |
yield self.url(page=file_name), file_name |
|
104 |
||
199 | 105 |
def page (self, name): |
106 |
""" |
|
107 |
Scans through tree looking for a matching page. |
|
108 |
||
109 |
Returns Page. |
|
110 |
""" |
|
200 | 111 |
|
112 |
if not name: |
|
113 |
name = self.INDEX |
|
199 | 114 |
|
200 | 115 |
for path, file_name, file_type in self.scan(): |
116 |
# match on name |
|
199 | 117 |
if file_name != name: |
118 |
continue |
|
119 |
||
200 | 120 |
# match on type |
121 |
if not file_type: |
|
122 |
continue |
|
123 |
||
199 | 124 |
page_type = TYPES.get(file_type) |
125 |
||
126 |
if not page_type: |
|
127 |
continue |
|
200 | 128 |
|
129 |
# out |
|
199 | 130 |
return page_type( |
131 |
path = path, |
|
132 |
name = name, |
|
133 |
tree = self, |
|
134 |
) |
|
135 |
||
197
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
136 |
class Page (object): |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
137 |
ENCODING = 'utf-8' |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
138 |
|
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
139 |
@classmethod |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
140 |
def lookup (cls, page): |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
141 |
""" |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
142 |
Lookup a Page from disk. |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
143 |
|
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
144 |
Returns None if there is no such page. |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
145 |
""" |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
146 |
|
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
147 |
log.info("page=%r", page) |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
148 |
|
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
149 |
if page: |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
150 |
parts = page.split('/') |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
151 |
else: |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
152 |
parts = [ ] |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
153 |
|
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
154 |
if parts: |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
155 |
page_name = parts.pop(-1) |
199 | 156 |
tree_parts = parts |
197
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
157 |
else: |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
158 |
page_name = '' |
199 | 159 |
tree_parts = [] |
197
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
160 |
|
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
161 |
# scan dir |
199 | 162 |
tree = Tree.lookup(tree_parts) |
197
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
163 |
|
199 | 164 |
if not tree: |
197
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
165 |
return None |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
166 |
|
199 | 167 |
# scan page |
168 |
page = tree.page(page_name) |
|
197
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
169 |
|
199 | 170 |
if not page: |
171 |
return None |
|
172 |
||
173 |
return page |
|
174 |
||
175 |
def __init__ (self, path, name, tree, encoding=ENCODING): |
|
197
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
176 |
self.path = path |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
177 |
self.name = name |
199 | 178 |
self.tree = tree |
197
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
179 |
self.encoding = encoding |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
180 |
|
201 | 181 |
def url (self): |
182 |
return self.tree.url(page=self.name) |
|
183 |
||
197
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
184 |
def open (self): |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
185 |
return codecs.open(self.path, encoding=self.encoding) |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
186 |
|
198 | 187 |
def breadcrumb (self): |
201 | 188 |
for path, name in self.tree.breadcrumb(): |
189 |
yield path, name |
|
200 | 190 |
|
191 |
if self.name != self.tree.INDEX: |
|
201 | 192 |
yield self.url(), self.name |
198 | 193 |
|
197
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
194 |
def render (self, request): |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
195 |
raise NotImplementedError() |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
196 |
|
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
197 |
class HTML_Page (Page): |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
198 |
def render (self, request): |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
199 |
return self.open().read() |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
200 |
|
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
201 |
TYPES = { |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
202 |
'html': HTML_Page, |
11a6d8fea463
qmsk_www_pages: pages.Page.lookup() and .html support; logging
Tero Marttila <terom@paivola.fi>
parents:
diff
changeset
|
203 |
} |