Mercurial > hg > index.fcgi > www > www-1

annotate pics2/pics_app.py @ 71:6318de36e334

find changesets by author, revision, files, or words in the commit message
pics2: limit access to index
author paulo
date Wed, 28 Oct 2015 00:57:23 -0700
parents b7966ae653f2
children ec48011dca1e
rev   line source
paulo@52 1 import os
paulo@52 2 import re
paulo@52 3 import glob
paulo@52 4 import traceback
paulo@54 5 import datetime
paulo@54 6 import random
paulo@54 7 import urlparse
paulo@52 8
paulo@52 9 import html
paulo@52 10
paulo@53 11 import logging
paulo@53 12 logging.basicConfig(
paulo@71 13 level=logging.INFO,
paulo@53 14 filename="_LOG",
paulo@53 15 format="%(asctime)s %(levelname)-8s %(message)s",
paulo@53 16 )
paulo@52 17
paulo@52 18
paulo@71 19 def _lahat():
paulo@71 20 ret = None
paulo@71 21
paulo@71 22 try:
paulo@71 23 with open("_lahat") as lahat_f:
paulo@71 24 ret = lahat_f.read().strip()
paulo@71 25 except IOError:
paulo@71 26 pass
paulo@71 27
paulo@71 28 return ret
paulo@71 29
paulo@52 30
paulo@54 31 def _is_pics_dir(dirpath):
paulo@54 32 return os.path.exists(os.path.join(dirpath, "_picsroot"))
paulo@52 33
paulo@52 34
paulo@54 35 def _get_dir_dt(dirpath):
paulo@54 36 if _is_pics_dir(dirpath):
paulo@54 37 dirpath = os.path.join(dirpath, "_picsroot")
paulo@54 38
paulo@54 39 return datetime.datetime.fromtimestamp(os.stat(dirpath).st_mtime)
paulo@54 40
paulo@54 41
paulo@54 42 def _format_dt(dt):
paulo@54 43 return dt.strftime("%Y-%m-%d")
paulo@53 44
paulo@53 45
paulo@53 46 def _parse_path_info(path_info):
paulo@53 47 ppi = path_info.split(os.sep)
paulo@53 48 if len(ppi) > 1 and ppi[-1] == '':
paulo@53 49 del ppi[-1]
paulo@53 50
paulo@53 51 return ppi
paulo@53 52
paulo@53 53
paulo@53 54 def _numeric_pad_basename(path, maxdigits=20):
paulo@53 55 return os.path.basename(path).zfill(maxdigits)
paulo@53 56
paulo@53 57
paulo@53 58 def _get_images(d):
paulo@57 59 exts = [".jpg", ".webm"]
paulo@57 60
paulo@53 61 thumb_fns = glob.glob(os.path.join(d, "thumbs", "*.jpg"))
paulo@53 62 thumb_fns = sorted(thumb_fns, key=_numeric_pad_basename)
paulo@53 63 logging.debug("thumb_fns = %s" % thumb_fns)
paulo@53 64
paulo@57 65 browse_dir = os.path.join(d, "browse")
paulo@57 66 browse_contents = set(os.listdir(browse_dir))
paulo@57 67 logging.debug("browse_contents = %s" % browse_contents)
paulo@57 68
paulo@57 69 browse_fns = []
paulo@57 70 for i in thumb_fns:
paulo@57 71 i_basename = os.path.splitext(os.path.basename(i))[0]
paulo@57 72 try:
paulo@57 73 for j in exts:
paulo@57 74 browse_fn_basename = i_basename + j
paulo@57 75 if browse_fn_basename in browse_contents:
paulo@57 76 browse_fns.append(os.path.join(browse_dir, browse_fn_basename))
paulo@57 77 raise StopIteration
paulo@57 78 except StopIteration:
paulo@57 79 pass
paulo@57 80 else:
paulo@57 81 raise RuntimeError("Cannot find browse image for %s" % i)
paulo@53 82 logging.debug("browse_fns = %s" % browse_fns)
paulo@53 83
paulo@53 84 return zip(thumb_fns, browse_fns)
paulo@53 85
paulo@52 86
paulo@54 87 class Main:
paulo@54 88 def _get_pics_url(self, dirpath):
paulo@54 89 script_name = self._environ.get("SCRIPT_NAME", '')
paulo@54 90 return os.path.normpath(os.path.join(os.path.dirname(script_name), dirpath))
paulo@54 91
paulo@54 92
paulo@54 93 def _get_app_url(self, dirpath):
paulo@54 94 script_name = self._environ.get("SCRIPT_NAME", '')
paulo@54 95 return os.path.normpath(os.path.join(script_name, dirpath))
paulo@54 96
paulo@54 97
paulo@54 98 def _get_standard_html_doc(self, title):
paulo@54 99 root = html.HTML("html")
paulo@54 100
paulo@54 101 header = root.header
paulo@54 102 header.link(rel="stylesheet", type="text/css", href=self._get_pics_url("index.css"))
paulo@54 103 header.title(title)
paulo@54 104
paulo@54 105 body = root.body
paulo@54 106 body.h1(title)
paulo@54 107
paulo@54 108 return (root, header, body)
paulo@54 109
paulo@54 110
paulo@56 111 def _go_thumbnail_links_to_browse_imgs_html_body(self, body, t, b, a_args={}, img_args={}):
paulo@54 112 thumb_img_url = self._get_pics_url(t)
paulo@54 113 browse_url = self._get_app_url(b)
paulo@52 114
paulo@56 115 a = body.a(href=browse_url, **a_args)
paulo@56 116 a.img(src=thumb_img_url, **img_args)
paulo@52 117
paulo@54 118 body.text(' ')
paulo@54 119
paulo@57 120
paulo@57 121 def _go_browse_image_html_body(self, body, img):
paulo@57 122 browse_img_url = self._get_pics_url(img)
paulo@57 123
paulo@57 124 ext = os.path.splitext(img)[1]
paulo@57 125 if ext == ".webm":
paulo@57 126 body.video(src=browse_img_url, autoplay="true", loop="true")
paulo@57 127 else:
paulo@57 128 body.img(src=browse_img_url)
paulo@57 129
paulo@54 130
paulo@54 131 def __init__(self, environ):
paulo@54 132 self._environ = environ
paulo@54 133 self._page_func = None
paulo@71 134 self._show_index = False
paulo@54 135
paulo@54 136 #logging.debug("environ = %s" % (sorted(self._environ.items(), key=lambda x: x[0]),))
paulo@54 137 logging.debug("environ['PATH_INFO'] = %s" % self._environ["PATH_INFO"])
paulo@54 138 logging.debug("environ['SCRIPT_NAME'] = %s" % self._environ["SCRIPT_NAME"])
paulo@54 139 logging.debug("environ['QUERY_STRING'] = %s" % self._environ["QUERY_STRING"])
paulo@54 140
paulo@54 141 pi = self._environ["PATH_INFO"]
paulo@54 142 ppi = _parse_path_info(pi)
paulo@54 143 logging.debug("ppi = %s" % ppi)
paulo@54 144
paulo@54 145 if len(ppi) < 1 or ppi[0] != '':
paulo@54 146 raise AssertionError("Parsed path length must start empty: " + pi)
paulo@54 147
paulo@71 148 self._qs = urlparse.parse_qs(self._environ["QUERY_STRING"])
paulo@71 149 logging.debug("self._qs = %s" % self._qs)
paulo@71 150 if self._qs.get("lahat", [''])[0] == _lahat():
paulo@71 151 self._show_index = True
paulo@71 152
paulo@54 153 if len(ppi) >= 2 and _is_pics_dir(ppi[1]):
paulo@54 154 if len(ppi) == 2:
paulo@54 155 self._page_func = self.page_thumbs
paulo@54 156 elif len(ppi) >= 4 and ppi[2] == "browse" and os.path.exists(os.path.join(*ppi)):
paulo@54 157 self._page_func = self.page_browse
paulo@71 158 elif len(ppi) == 1 and self._show_index:
paulo@54 159 self._page_func = self.page_index
paulo@54 160
paulo@54 161 if self._page_func is None:
paulo@54 162 raise RuntimeError("Cannot find path: " + pi)
paulo@54 163
paulo@53 164
paulo@54 165 def page(self):
paulo@54 166 return unicode(self._page_func()).encode("utf-8")
paulo@54 167
paulo@53 168
paulo@54 169 def page_index(self):
paulo@54 170 n = 5 # number of thumbnails to display
paulo@53 171
paulo@54 172 (html_root, html_header, html_body) = self._get_standard_html_doc("Pictures")
paulo@54 173
paulo@54 174 pics_dirs = []
paulo@54 175 for i in os.listdir('.'):
paulo@54 176 if _is_pics_dir(i):
paulo@54 177 pics_dirs.append((i, _get_dir_dt(i)))
paulo@54 178
paulo@54 179 pics_dirs.sort(key=lambda x: x[1], reverse=True)
paulo@54 180
paulo@54 181 for (d, dt) in pics_dirs:
paulo@54 182 html_body.h2.a(d, href=self._get_app_url(d))
paulo@54 183 html_body.h3(_format_dt(dt))
paulo@54 184
paulo@54 185 imgs = _get_images(d)
paulo@54 186 imgs_idx = [(i, img) for (i, img) in enumerate(imgs)]
paulo@54 187
paulo@54 188 sampled_imgs_idx = random.sample(imgs_idx, min(len(imgs_idx), n))
paulo@54 189 sampled_imgs_idx.sort(key=lambda x: x[0])
paulo@54 190
paulo@54 191 html_p = html_body.p
paulo@54 192 for (i, (t, b)) in sampled_imgs_idx:
paulo@54 193 self._go_thumbnail_links_to_browse_imgs_html_body(html_p, t, b)
paulo@54 194
paulo@54 195 return html_root
paulo@54 196
paulo@54 197
paulo@54 198 def page_thumbs(self):
paulo@54 199 ppi = _parse_path_info(self._environ["PATH_INFO"])
paulo@54 200 d = os.path.join(*ppi)
paulo@54 201 (html_root, html_header, html_body) = self._get_standard_html_doc(d)
paulo@53 202
paulo@54 203 from_img = None
paulo@71 204 if "from" in self._qs:
paulo@71 205 from_img = self._qs["from"][0]
paulo@54 206
paulo@54 207 html_p = html_body.p
paulo@54 208 for (t, b) in _get_images(d):
paulo@54 209 if from_img is not None and b == from_img:
paulo@56 210 self._go_thumbnail_links_to_browse_imgs_html_body(html_p, t, b, img_args={"klass":"sel2", "id":"selected"})
paulo@54 211 else:
paulo@54 212 self._go_thumbnail_links_to_browse_imgs_html_body(html_p, t, b)
paulo@54 213
paulo@71 214 # TODO: fix me
paulo@71 215 #if self._show_index:
paulo@71 216 # html_body.a("(Other pictures)", href=self._get_app_url(''))
paulo@71 217
paulo@54 218 return html_root
paulo@54 219
paulo@54 220
paulo@54 221 def page_browse(self):
paulo@54 222 ppi = _parse_path_info(self._environ["PATH_INFO"])
paulo@54 223 d = os.path.join(*ppi[:2])
paulo@54 224 imgs = _get_images(d)
paulo@54 225 img = os.path.join(*ppi)
paulo@54 226
paulo@54 227 # thumbnail preview ribbon
paulo@54 228 w = 7 # must be odd
paulo@54 229 v = w/2
paulo@54 230 imgs_circ = [None] * w
paulo@54 231 x = None
paulo@54 232 n = len(imgs)
paulo@54 233 for (i, (t, b)) in enumerate(imgs):
paulo@54 234 if b == img:
paulo@54 235 x = i + 1
paulo@54 236 imgs_circ[v] = (t, b)
paulo@54 237 for j in range(1, v + 1):
paulo@54 238 if (i + j) < n: imgs_circ[v + j] = imgs[i + j]
paulo@54 239 if (i - j) >= 0: imgs_circ[v - j] = imgs[i - j]
paulo@54 240
paulo@54 241 break
paulo@54 242
paulo@54 243 if x is None:
paulo@54 244 raise AssertionError
paulo@54 245
paulo@54 246 (html_root, html_header, html_body) = self._get_standard_html_doc(u"%s \u2014 %s of %s" % (d, x, len(imgs)))
paulo@56 247
paulo@56 248 html_header.script('', type="text/javascript", src=self._get_pics_url("np_keys.js"))
paulo@54 249
paulo@57 250 self._go_browse_image_html_body(html_body.p, img)
paulo@54 251
paulo@54 252 logging.debug("imgs_circ = %s" % imgs_circ)
paulo@54 253
paulo@54 254 html_p = html_body.p
paulo@56 255 for (i, img_c) in enumerate(imgs_circ):
paulo@56 256 if img_c is not None:
paulo@56 257 (t, b) = img_c
paulo@56 258 a_args = {}
paulo@56 259 img_args = {}
paulo@54 260 if b == img:
paulo@56 261 a_args = {"id": "up"}
paulo@56 262 img_args = {"klass": "sel"}
paulo@56 263 b = "%s?from=%s#selected" % (d, img)
paulo@56 264 elif i == v + 1:
paulo@56 265 a_args = {"id": "next"}
paulo@56 266 elif i == v - 1:
paulo@56 267 a_args = {"id": "prev"}
paulo@56 268
paulo@56 269 self._go_thumbnail_links_to_browse_imgs_html_body(html_p, t, b, a_args, img_args)
paulo@54 270
paulo@54 271 return html_root
paulo@53 272
paulo@52 273
paulo@52 274 def app(environ, start_response):
paulo@52 275 response_code = "500 Internal Server Error"
paulo@52 276 response_type = "text/plain; charset=UTF-8"
paulo@52 277
paulo@52 278 try:
paulo@54 279 response_body = Main(environ).page()
paulo@52 280 response_code = "200 OK"
paulo@52 281 response_type = "text/html; charset=UTF-8"
paulo@52 282 except:
paulo@52 283 response_body = traceback.format_exc()
paulo@52 284
paulo@52 285 response_headers = [
paulo@52 286 ("Content-Type", response_type),
paulo@52 287 ("Content-Length", str(len(response_body))),
paulo@52 288 ]
paulo@52 289
paulo@52 290 start_response(response_code, response_headers)
paulo@52 291
paulo@52 292 return [response_body]