mirror of
https://github.com/postgres/pgweb.git
synced 2025-08-03 15:38:59 +00:00
52 lines
1.9 KiB
Python
Executable File
52 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# This hook script is run by gitdeployer when the code is
|
|
# deployed to a server. figure out which templates have been
|
|
# modified and queue automatic purges for them.
|
|
# Gitdeployer will pass a list of all modified files on
|
|
# stdin.
|
|
|
|
import sys
|
|
import os
|
|
import hashlib
|
|
from configparser import ConfigParser
|
|
import psycopg2
|
|
|
|
# Templates that we don't want to ban automatically
|
|
BANNED_TEMPLATES = (
|
|
'base/base.html',
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
config = ConfigParser()
|
|
config.read(os.path.join(os.path.abspath(os.path.dirname(__file__)), 'purgehook.ini'))
|
|
conn = psycopg2.connect(config.get('db', 'dsn'))
|
|
curs = conn.cursor()
|
|
|
|
for l in sys.stdin:
|
|
if '--static' in sys.argv:
|
|
# For the static files part, we always purge just on filename, with a prefix
|
|
curs.execute("SELECT varnish_purge('^/files/' || %(u)s || '$')", {
|
|
'u': l.strip(),
|
|
})
|
|
|
|
# If the purge is for documentation PDF, also purge the xkey for docs pdfs,
|
|
# so we update the list of which are available.
|
|
if l.startswith('documentation/pdf/'):
|
|
curs.execute("SELECT varnish_purge_xkey('pgdocs_pdf')")
|
|
elif l.startswith('templates/'):
|
|
# On regular website, if it's a template do an xkey purge of all pages using that template
|
|
tmpl = l[len('templates/'):].strip()
|
|
if tmpl not in BANNED_TEMPLATES:
|
|
curs.execute("SELECT varnish_purge_xkey(%(key)s)", {
|
|
'key': 'pgwt_{0}'.format(hashlib.md5(tmpl.encode('ascii')).hexdigest()),
|
|
})
|
|
elif l.startswith('media/'):
|
|
# For media we can't xkey, but the URL is exact so we can
|
|
# use a classic single-url purge.
|
|
curs.execute("SELECT varnish_purge('^/' || %(u)s || '$')", {
|
|
'u': l.strip(),
|
|
})
|
|
conn.commit()
|
|
conn.close()
|