My static website generator using poole https://www.xythobuz.de
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

macros.py 22KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622
  1. # -*- coding: utf-8 -*-
  2. import re
  3. import itertools
  4. import email.utils
  5. import os.path
  6. import time
  7. import codecs
  8. from datetime import datetime
  9. DEFAULT_LANG = "en"
  10. BASE_URL = "https://www.xythobuz.de"
  11. # -----------------------------------------------------------------------------
  12. # sub page helper macro
  13. # -----------------------------------------------------------------------------
  14. def backToParent():
  15. url = page.get("parent", "") + ".html"
  16. posts = [p for p in pages if p.url == url]
  17. if len(posts) > 0:
  18. p = posts[0]
  19. print '<span class="listdesc">[...back to ' + p.title + ' overview](' + p.url + ')</span>'
  20. # -----------------------------------------------------------------------------
  21. # table helper macro
  22. # -----------------------------------------------------------------------------
  23. def tableHelper(style, header, content):
  24. print "<table>"
  25. if (header != None) and (len(header) == len(style)):
  26. print "<tr>"
  27. for h in header:
  28. print "<th>" + h + "</th>"
  29. print "</tr>"
  30. for ci in range(0, len(content)):
  31. if len(content[ci]) != len(style):
  32. # invalid call of table helper!
  33. continue
  34. print "<tr>"
  35. for i in range(0, len(style)):
  36. s = style[i]
  37. td_style = ""
  38. if "monospaced" in s:
  39. td_style += " font-family: monospace;"
  40. if "align-last-right" in s:
  41. if ci == (len(content) - 1):
  42. td_style += " text-align: right;"
  43. else:
  44. if "align-center" in s:
  45. td_style += " text-align: center;"
  46. elif "align-right" in s:
  47. td_style += " text-align: right;"
  48. elif "align-center" in s:
  49. td_style += " text-align: center;"
  50. td_args = ""
  51. if td_style != "":
  52. td_args = " style=\"" + td_style + "\""
  53. print "<td" + td_args + ">"
  54. if isinstance(content[ci][i], tuple):
  55. text, link = content[ci][i]
  56. print "<a href=\"" + link + "\">" + text + "</a>"
  57. else:
  58. text = content[ci][i]
  59. print text
  60. print "</td>"
  61. print "</tr>"
  62. print "</table>"
  63. # -----------------------------------------------------------------------------
  64. # menu helper macro
  65. # -----------------------------------------------------------------------------
  66. def githubCommitBadge(p, showInline = False):
  67. ret = ""
  68. if p.get("github", "") != "":
  69. link = p.get("git", p.github)
  70. linkParts = p.github.split("/")
  71. if len(linkParts) >= 5:
  72. ret += "<a href=\"" + link + "\"><img "
  73. if showInline:
  74. ret += "style =\"vertical-align: top;\" "
  75. ret += "src=\"https://img.shields.io/github/last-commit/"
  76. ret += linkParts[3] + "/" + linkParts[4]
  77. ret += ".svg?logo=git&style=flat\" /></a>"
  78. return ret
  79. def printMenuItem(p, yearsAsHeading = False, showDateSpan = False, showOnlyStartDate = False, nicelyFormatFullDate = False, lastyear = "0", lang = "", showLastCommit = True):
  80. title = p.title
  81. if lang != "":
  82. if p.get("title_" + lang, "") != "":
  83. title = p.get("title_" + lang, "")
  84. if title == "Blog":
  85. title = p.post
  86. year = p.get("date", "")[0:4]
  87. if year != lastyear:
  88. lastyear = year
  89. if yearsAsHeading:
  90. print "\n\n#### %s\n" % (year)
  91. dateto = ""
  92. if p.get("date", "" != ""):
  93. year = p.get("date", "")[0:4]
  94. if showOnlyStartDate:
  95. dateto = " (%s)" % (year)
  96. if p.get("update", "") != "" and p.get("update", "")[0:4] != year:
  97. if showDateSpan:
  98. dateto = " (%s - %s)" % (year, p.get("update", "")[0:4])
  99. if nicelyFormatFullDate:
  100. dateto = " - " + datetime.strptime(p.get("update", p.date), "%Y-%m-%d").strftime("%B %d, %Y")
  101. print " * **[%s](%s)**%s" % (title, p.url, dateto)
  102. if p.get("description", "") != "":
  103. description = p.get("description", "")
  104. if lang != "":
  105. if p.get("description_" + lang, "") != "":
  106. description = p.get("description_" + lang, "")
  107. print "<br><span class=\"listdesc\">" + description + "</span>"
  108. if showLastCommit:
  109. link = githubCommitBadge(p)
  110. if len(link) > 0:
  111. print "<br>" + link
  112. return lastyear
  113. def printRecentMenu(count = 5):
  114. posts = [p for p in pages if "date" in p and p.lang == "en"]
  115. posts.sort(key=lambda p: p.get("update", p.get("date")), reverse=True)
  116. if count > 0:
  117. posts = posts[0:count]
  118. for p in posts:
  119. printMenuItem(p, False, False, False, True, "0", "", False)
  120. def printBlogMenu():
  121. posts = [p for p in pages if "post" in p and p.lang == "en"]
  122. posts.sort(key=lambda p: p.get("date", "9999-01-01"), reverse=True)
  123. lastyear = "0"
  124. for p in posts:
  125. lastyear = printMenuItem(p, True, False, False, True, lastyear)
  126. def printProjectsMenu():
  127. # prints all pages with parent 'projects' or 'stuff'.
  128. # first the ones without date, sorted by position.
  129. # then afterwards those with date, split by year.
  130. # also supports blog posts with parent.
  131. enpages = [p for p in pages if p.lang == "en"]
  132. dpages = [p for p in enpages if p.get("date", "") == ""]
  133. mpages = [p for p in dpages if any(x in p.get("parent", "") for x in [ 'projects', 'stuff' ])]
  134. mpages.sort(key=lambda p: [int(p.get("position", "999"))])
  135. for p in mpages:
  136. printMenuItem(p)
  137. dpages = [p for p in enpages if p.get("date", "") != ""]
  138. mpages = [p for p in dpages if any(x in p.get("parent", "") for x in [ 'projects', 'stuff' ])]
  139. mpages.sort(key=lambda p: [p.get("date", "9999-01-01")], reverse = True)
  140. lastyear = "0"
  141. for p in mpages:
  142. lastyear = printMenuItem(p, True, True, False, False, lastyear)
  143. def print3DPrintingMenu():
  144. mpages = [p for p in pages if p.get("parent", "") == "3d-printing" and p.lang == "en"]
  145. mpages.sort(key=lambda p: int(p["position"]))
  146. for p in mpages:
  147. printMenuItem(p, False, True, True)
  148. def printSmarthomeMenu():
  149. mpages = [p for p in pages if p.get("parent", "") == "smarthome" and p.lang == "en"]
  150. mpages.sort(key=lambda p: int(p["position"]))
  151. for p in mpages:
  152. printMenuItem(p, False, True, True)
  153. def printQuadcopterMenu():
  154. mpages = [p for p in pages if p.get("parent", "") == "quadcopters" and p.lang == "en"]
  155. mpages.sort(key=lambda p: int(p["position"]))
  156. for p in mpages:
  157. printMenuItem(p, False, True, True)
  158. def printQuadcopterRelatedMenu():
  159. mpages = [p for p in pages if p.get("show_in_quadcopters", "false") == "true"]
  160. mpages.sort(key=lambda p: [p.get("date", "9999-01-01")], reverse = True)
  161. for p in mpages:
  162. printMenuItem(p, False, True, True)
  163. def printRobotMenuEnglish():
  164. mpages = [p for p in pages if p.get("parent", "") == "xyrobot" and p.lang == "en"]
  165. mpages.sort(key=lambda p: int(p["position"]))
  166. for p in mpages:
  167. printMenuItem(p)
  168. def printRobotMenuDeutsch():
  169. mpages = [p for p in pages if p.get("parent", "") == "xyrobot" and p.lang == "de"]
  170. mpages.sort(key=lambda p: int(p["position"]))
  171. for p in mpages:
  172. printMenuItem(p, False, False, False, False, "0", "de")
  173. # -----------------------------------------------------------------------------
  174. # lightgallery helper macro
  175. # -----------------------------------------------------------------------------
  176. # call this macro like this:
  177. # lightgallery([
  178. # [ "image-link", "description" ],
  179. # [ "image-link", "thumbnail-link", "description" ],
  180. # [ "youtube-link", "thumbnail-link", "description" ],
  181. # [ "video-link", "mime", "thumbnail-link", "image-link", "description" ]
  182. # ])
  183. # it will also auto-generate thumbnails and resize and strip EXIF from images
  184. # using the included web-image-resize script.
  185. def lightgallery_check_thumbnail(link, thumb):
  186. # only check local image links
  187. if not link.startswith('img/'):
  188. return
  189. # generate thumbnail filename web-image-resize will create
  190. x = link.rfind('.')
  191. img = link[:x] + '_small' + link[x:]
  192. # only run when desired thumb path matches calculated ones
  193. if thumb != img:
  194. return
  195. # generate fs path to images
  196. path = os.path.join(os.getcwd(), 'static', link)
  197. img = os.path.join(os.getcwd(), 'static', thumb)
  198. # no need to generate thumb again
  199. if os.path.exists(img):
  200. return
  201. # run web-image-resize to generate thumbnail
  202. script = os.path.join(os.getcwd(), 'web-image-resize')
  203. os.system(script + ' ' + path)
  204. def lightgallery(links):
  205. videos = [l for l in links if len(l) == 5]
  206. v_i = -1
  207. for v in videos:
  208. link, mime, thumb, poster, alt = v
  209. v_i += 1
  210. print '<div style="display:none;" id="video' + str(v_i) + '">'
  211. print '<video class="lg-video-object lg-html5" controls preload="none">'
  212. print '<source src="' + link + '" type="' + mime + '">'
  213. print 'Your browser does not support HTML5 video.'
  214. print '</video>'
  215. print '</div>'
  216. print '<div class="lightgallery">'
  217. v_i = -1
  218. for l in links:
  219. if (len(l) == 3) or (len(l) == 2):
  220. link = img = alt = ""
  221. if len(l) == 3:
  222. link, img, alt = l
  223. else:
  224. link, alt = l
  225. x = link.rfind('.')
  226. img = link[:x] + '_small' + link[x:]
  227. lightgallery_check_thumbnail(link, img)
  228. print '<div class="border" data-src="' + link + '"><a href="' + link + '"><img class="pic" src="' + img + '" alt="' + alt + '"></a></div>'
  229. elif len(l) == 5:
  230. v_i += 1
  231. link, mime, thumb, poster, alt = videos[v_i]
  232. print '<div class="border" data-poster="' + poster + '" data-sub-html="' + alt + '" data-html="#video' + str(v_i) + '"><a href="' + link + '"><img class="pic" src="' + thumb + '"></a></div>'
  233. else:
  234. raise NameError('Invalid number of arguments for lightgallery')
  235. print '</div>'
  236. # -----------------------------------------------------------------------------
  237. # github helper macros
  238. # -----------------------------------------------------------------------------
  239. import urllib, json
  240. def restRequest(url):
  241. response = urllib.urlopen(url)
  242. data = json.loads(response.read())
  243. return data
  244. def restReleases(user, repo):
  245. s = "https://api.github.com/repos/"
  246. s += user
  247. s += "/"
  248. s += repo
  249. s += "/releases"
  250. return restRequest(s)
  251. def printLatestRelease(user, repo):
  252. repo_url = "https://github.com/" + user + "/" + repo
  253. print("<div class=\"releasecard\">")
  254. print("Release builds for " + repo + " are <a href=\"" + repo_url + "/releases\">available on GitHub</a>.<br>\n")
  255. releases = restReleases(user, repo)
  256. if len(releases) <= 0:
  257. print("No release has been published on GitHub yet.")
  258. print("</div>")
  259. return
  260. releases.sort(key=lambda x: x["published_at"], reverse=True)
  261. r = releases[0]
  262. release_url = r["html_url"]
  263. print("Latest release of <a href=\"" + repo_url + "\">" + repo + "</a>, at the time of this writing: <a href=\"" + release_url + "\">" + r["name"] + "</a> (" + datetime.strptime(r["published_at"], "%Y-%m-%dT%H:%M:%SZ").strftime("%Y-%m-%d %H:%M:%S") + ")\n")
  264. if len(r["assets"]) <= 0:
  265. print("<br>No release assets have been published on GitHub for that.")
  266. print("</div>")
  267. return
  268. print("<ul>")
  269. print("Release Assets:")
  270. for a in r["assets"]:
  271. size = int(a["size"])
  272. ss = " "
  273. if size >= (1024 * 1024):
  274. ss += "(%.1f MiB)" % (size / (1024.0 * 1024.0))
  275. elif size >= 1024:
  276. ss += "(%d KiB)" % (size // 1024)
  277. else:
  278. ss += "(%d Byte)" % (size)
  279. print("<li><a href=\"" + a["browser_download_url"] + "\">" + a["name"] + "</a>" + ss)
  280. print("</ul></div>")
  281. # -----------------------------------------------------------------------------
  282. # preconvert hooks
  283. # -----------------------------------------------------------------------------
  284. # -----------------------------------------------------------------------------
  285. # multi language support
  286. # -----------------------------------------------------------------------------
  287. def hook_preconvert_anotherlang():
  288. MKD_PATT = r'\.(?:md|mkd|mdown|markdown)$'
  289. _re_lang = re.compile(r'^[\s+]?lang[\s+]?[:=]((?:.|\n )*)', re.MULTILINE)
  290. vpages = [] # Set of all virtual pages
  291. for p in pages:
  292. current_lang = DEFAULT_LANG # Default language
  293. langs = [] # List of languages for the current page
  294. page_vpages = {} # Set of virtual pages for the current page
  295. text_lang = re.split(_re_lang, p.source)
  296. text_grouped = dict(zip([current_lang,] + \
  297. [lang.strip() for lang in text_lang[1::2]], \
  298. text_lang[::2]))
  299. for lang, text in text_grouped.iteritems():
  300. spath = p.fname.split(os.path.sep)
  301. langs.append(lang)
  302. if lang == "en":
  303. filename = re.sub(MKD_PATT, "%s\g<0>" % "", p.fname).split(os.path.sep)[-1]
  304. else:
  305. filename = re.sub(MKD_PATT, ".%s\g<0>" % lang, p.fname).split(os.path.sep)[-1]
  306. vp = Page(filename, virtual=text)
  307. # Copy real page attributes to the virtual page
  308. for attr in p:
  309. if not vp.has_key(attr):
  310. vp[attr] = p[attr]
  311. # Define a title in the proper language
  312. vp["title"] = p["title_%s" % lang] \
  313. if p.has_key("title_%s" % lang) \
  314. else p["title"]
  315. # Keep track of the current lang of the virtual page
  316. vp["lang"] = lang
  317. page_vpages[lang] = vp
  318. # Each virtual page has to know about its sister vpages
  319. for lang, vpage in page_vpages.iteritems():
  320. vpage["lang_links"] = dict([(l, v["url"]) for l, v in page_vpages.iteritems()])
  321. vpage["other_lang"] = langs # set other langs and link
  322. vpages += page_vpages.values()
  323. pages[:] = vpages
  324. # -----------------------------------------------------------------------------
  325. # compatibility redirect for old website URLs
  326. # -----------------------------------------------------------------------------
  327. _COMPAT = """ case "%s":
  328. $loc = "%s/%s";
  329. break;
  330. """
  331. _COMPAT_404 = """ default:
  332. $loc = "%s";
  333. break;
  334. """
  335. def hook_preconvert_compat():
  336. fp = open(os.path.join(options.project, "output", "index.php"), 'w')
  337. fp.write("<?\n")
  338. fp.write("// Auto generated xyCMS compatibility index.php\n")
  339. fp.write("$loc = 'https://www.xythobuz.de/index.de.html';\n")
  340. fp.write("if (isset($_GET['p'])) {\n")
  341. fp.write(" if (isset($_GET['lang'])) {\n")
  342. fp.write(" $_GET['p'] .= 'EN';\n")
  343. fp.write(" }\n")
  344. fp.write(" switch($_GET['p']) {\n")
  345. for p in pages:
  346. if p.get("compat", "") != "":
  347. tmp = p["compat"]
  348. if p.get("lang", DEFAULT_LANG) == DEFAULT_LANG:
  349. tmp = tmp + "EN"
  350. fp.write(_COMPAT % (tmp, "https://www.xythobuz.de", p.url))
  351. fp.write("\n")
  352. fp.write(_COMPAT_404 % "/404.html")
  353. fp.write(" }\n")
  354. fp.write("}\n")
  355. fp.write("if ($_SERVER['SERVER_PROTOCOL'] == 'HTTP/1.1') {\n")
  356. fp.write(" if (php_sapi_name() == 'cgi') {\n")
  357. fp.write(" header('Status: 301 Moved Permanently');\n")
  358. fp.write(" } else {\n")
  359. fp.write(" header('HTTP/1.1 301 Moved Permanently');\n")
  360. fp.write(" }\n")
  361. fp.write("}\n");
  362. fp.write("header('Location: '.$loc);\n")
  363. fp.write("?>")
  364. fp.close()
  365. # -----------------------------------------------------------------------------
  366. # sitemap generation
  367. # -----------------------------------------------------------------------------
  368. _SITEMAP = """<?xml version="1.0" encoding="UTF-8"?>
  369. <urlset xmlns="http://www.sitemaps.org/schemas/sitemap/0.9">
  370. %s
  371. </urlset>
  372. """
  373. _SITEMAP_URL = """
  374. <url>
  375. <loc>%s/%s</loc>
  376. <lastmod>%s</lastmod>
  377. <changefreq>%s</changefreq>
  378. <priority>%s</priority>
  379. </url>
  380. """
  381. def hook_preconvert_sitemap():
  382. date = datetime.strftime(datetime.now(), "%Y-%m-%d")
  383. urls = []
  384. for p in pages:
  385. urls.append(_SITEMAP_URL % (BASE_URL, p.url, date, p.get("changefreq", "monthly"), p.get("priority", "0.5")))
  386. fname = os.path.join(options.project, "output", "sitemap.xml")
  387. fp = open(fname, 'w')
  388. fp.write(_SITEMAP % "".join(urls))
  389. fp.close()
  390. # -----------------------------------------------------------------------------
  391. # postconvert hooks
  392. # -----------------------------------------------------------------------------
  393. # -----------------------------------------------------------------------------
  394. # rss feed generation
  395. # -----------------------------------------------------------------------------
  396. _RSS = """<?xml version="1.0" encoding="UTF-8"?>
  397. <rss version="2.0" xmlns:atom="http://www.w3.org/2005/Atom">
  398. <channel>
  399. <title>%s</title>
  400. <link>%s</link>
  401. <atom:link href="%s" rel="self" type="application/rss+xml" />
  402. <description>%s</description>
  403. <language>en-us</language>
  404. <pubDate>%s</pubDate>
  405. <lastBuildDate>%s</lastBuildDate>
  406. <docs>http://blogs.law.harvard.edu/tech/rss</docs>
  407. <generator>Poole</generator>
  408. <ttl>720</ttl>
  409. %s
  410. </channel>
  411. </rss>
  412. """
  413. _RSS_ITEM = """
  414. <item>
  415. <title>%s</title>
  416. <link>%s</link>
  417. <description>%s</description>
  418. <pubDate>%s</pubDate>
  419. <atom:updated>%s</atom:updated>
  420. <guid>%s</guid>
  421. </item>
  422. """
  423. def hook_postconvert_rss():
  424. items = []
  425. # all pages with "date" get put into feed
  426. posts = [p for p in pages if "date" in p]
  427. # sort by update if available, date else
  428. posts.sort(key=lambda p: p.get("update", p.date), reverse=True)
  429. # only put 20 most recent items in feed
  430. posts = posts[:20]
  431. for p in posts:
  432. title = p.title
  433. if "post" in p:
  434. title = p.post
  435. link = "%s/%s" % (BASE_URL, p.url)
  436. desc = p.html.replace("href=\"img", "%s%s%s" % ("href=\"", BASE_URL, "/img"))
  437. desc = desc.replace("src=\"img", "%s%s%s" % ("src=\"", BASE_URL, "/img"))
  438. desc = desc.replace("href=\"/img", "%s%s%s" % ("href=\"", BASE_URL, "/img"))
  439. desc = desc.replace("src=\"/img", "%s%s%s" % ("src=\"", BASE_URL, "/img"))
  440. desc = htmlspecialchars(desc)
  441. date = time.mktime(time.strptime("%s 12" % p.date, "%Y-%m-%d %H"))
  442. date = email.utils.formatdate(date)
  443. update = time.mktime(time.strptime("%s 12" % p.get("update", p.date), "%Y-%m-%d %H"))
  444. update = email.utils.formatdate(update)
  445. items.append(_RSS_ITEM % (title, link, desc, date, update, link))
  446. items = "".join(items)
  447. title = "xythobuz.de Blog"
  448. link = "%s" % BASE_URL
  449. feed = "%s/rss.xml" % BASE_URL
  450. desc = htmlspecialchars("xythobuz Electronics & Software Projects")
  451. date = email.utils.formatdate()
  452. rss = _RSS % (title, link, feed, desc, date, date, items)
  453. fp = codecs.open(os.path.join(output, "rss.xml"), "w", "utf-8")
  454. fp.write(rss)
  455. fp.close()
  456. # -----------------------------------------------------------------------------
  457. # compatibility redirect for old mobile pages
  458. # -----------------------------------------------------------------------------
  459. _COMPAT_MOB = """ case "%s":
  460. $loc = "%s/%s";
  461. break;
  462. """
  463. _COMPAT_404_MOB = """ default:
  464. $loc = "%s";
  465. break;
  466. """
  467. def hook_postconvert_mobilecompat():
  468. directory = os.path.join(output, "mobile")
  469. if not os.path.exists(directory):
  470. os.makedirs(directory)
  471. fp = codecs.open(os.path.join(directory, "index.php"), "w", "utf-8")
  472. fp.write("<?\n")
  473. fp.write("// Auto generated xyCMS compatibility mobile/index.php\n")
  474. fp.write("$loc = 'https://www.xythobuz.de/index.de.html';\n")
  475. fp.write("if (isset($_GET['p'])) {\n")
  476. fp.write(" if (isset($_GET['lang'])) {\n")
  477. fp.write(" $_GET['p'] .= 'EN';\n")
  478. fp.write(" }\n")
  479. fp.write(" switch($_GET['p']) {\n")
  480. for p in pages:
  481. if p.get("compat", "") != "":
  482. tmp = p["compat"]
  483. if p.get("lang", DEFAULT_LANG) == DEFAULT_LANG:
  484. tmp = tmp + "EN"
  485. fp.write(_COMPAT_MOB % (tmp, "https://www.xythobuz.de", re.sub(".html", ".html", p.url)))
  486. fp.write("\n")
  487. fp.write(_COMPAT_404_MOB % "/404.mob.html")
  488. fp.write(" }\n")
  489. fp.write("}\n")
  490. fp.write("if ($_SERVER['SERVER_PROTOCOL'] == 'HTTP/1.1') {\n")
  491. fp.write(" if (php_sapi_name() == 'cgi') {\n")
  492. fp.write(" header('Status: 301 Moved Permanently');\n")
  493. fp.write(" } else {\n")
  494. fp.write(" header('HTTP/1.1 301 Moved Permanently');\n")
  495. fp.write(" }\n")
  496. fp.write("}\n");
  497. fp.write("header('Location: '.$loc);\n")
  498. fp.write("?>")
  499. fp.close()
  500. # -----------------------------------------------------------------------------
  501. # displaying filesize for download links
  502. # -----------------------------------------------------------------------------
  503. def hook_postconvert_size():
  504. file_ext = '|'.join(['pdf', 'zip', 'rar', 'ods', 'odt', 'odp', 'doc', 'xls', 'ppt', 'docx', 'xlsx', 'pptx', 'exe', 'brd', 'plist'])
  505. def matched_link(matchobj):
  506. try:
  507. path = matchobj.group(1)
  508. if path.startswith("http") or path.startswith("//") or path.startswith("ftp"):
  509. return '<a href=\"%s\">%s</a>' % (matchobj.group(1), matchobj.group(3))
  510. elif path.startswith("/"):
  511. path = path.strip("/")
  512. path = os.path.join("static/", path)
  513. size = os.path.getsize(path)
  514. if size >= (1024 * 1024):
  515. return "<a href=\"%s\">%s</a>&nbsp;(%.1f MiB)" % (matchobj.group(1), matchobj.group(3), size / (1024.0 * 1024.0))
  516. elif size >= 1024:
  517. return "<a href=\"%s\">%s</a>&nbsp;(%d KiB)" % (matchobj.group(1), matchobj.group(3), size // 1024)
  518. else:
  519. return "<a href=\"%s\">%s</a>&nbsp;(%d Byte)" % (matchobj.group(1), matchobj.group(3), size)
  520. except:
  521. print "Unable to estimate file size for %s" % matchobj.group(1)
  522. return '<a href=\"%s\">%s</a>' % (matchobj.group(1), matchobj.group(3))
  523. _re_url = '<a href=\"([^\"]*?\.(%s))\">(.*?)<\/a>' % file_ext
  524. for p in pages:
  525. p.html = re.sub(_re_url, matched_link, p.html)