mirror of
https://github.com/postgres/pgweb.git
synced 2025-07-25 16:02:27 +00:00
bcce9e0aee
Using commandline --quiet. To do this, switch to using the standard python optionparser for commandline parsing.
107 lines
2.7 KiB
Python
Executable File
107 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
# Script to load documentation from tarballs
|
|
|
|
import sys
|
|
import os
|
|
import tarfile
|
|
import re
|
|
import tidy
|
|
from optparse import OptionParser
|
|
|
|
|
|
# Set up for accessing django
|
|
from django.core.management import setup_environ
|
|
sys.path.append(os.path.join(os.path.abspath(os.path.dirname(sys.argv[0])), '../../pgweb'))
|
|
import settings
|
|
setup_environ(settings)
|
|
|
|
from django.db import connection, transaction
|
|
|
|
pagecount = 0
|
|
quiet = False
|
|
|
|
re_titlematch = re.compile('<title\s*>([^<]+)</title\s*>', re.IGNORECASE)
|
|
|
|
## Load a single page
|
|
def load_doc_file(filename, f):
|
|
tidyopts = dict(drop_proprietary_attributes=1,
|
|
alt_text='',
|
|
hide_comments=1,
|
|
output_xhtml=1,
|
|
show_body_only=1,
|
|
clean=1,
|
|
char_encoding='utf8',
|
|
indent='auto',
|
|
)
|
|
|
|
contents = unicode(f.read(),'latin1')
|
|
tm = re_titlematch.search(contents)
|
|
if tm:
|
|
title = tm.group(1)
|
|
else:
|
|
title = ""
|
|
if not quiet: print "--- file: %s (%s) ---" % (filename, title)
|
|
|
|
s = tidy.parseString(contents.encode('utf-8'), **tidyopts)
|
|
curs.execute("INSERT INTO docs (file, version, title, content) VALUES (%(f)s, %(v)s, %(t)s, %(c)s)",{
|
|
'f': filename,
|
|
'v': ver,
|
|
't': title,
|
|
'c': str(s),
|
|
})
|
|
global pagecount
|
|
pagecount += 1
|
|
|
|
## Main execution
|
|
|
|
parser = OptionParser(usage="usage: %prog [options] <version> <tarfile>")
|
|
parser.add_option("-q", "--quiet", action="store_true", dest="quiet",
|
|
help="Run quietly")
|
|
(options, args) = parser.parse_args()
|
|
|
|
if len(args) != 2:
|
|
parser.print_usage()
|
|
sys.exit(1)
|
|
|
|
quiet = options.quiet
|
|
ver = sys.argv[1]
|
|
tarfilename = sys.argv[2]
|
|
|
|
if not os.path.isfile(tarfilename):
|
|
print "File %s not found" % tarfilename
|
|
sys.exit(1)
|
|
|
|
tf = tarfile.open(tarfilename)
|
|
|
|
curs = connection.cursor()
|
|
# Verify that the version exists, and what we're loading
|
|
curs.execute("SELECT latestminor FROM core_version WHERE tree=%(v)s", {'v': ver})
|
|
r = curs.fetchall()
|
|
if len(r) != 1:
|
|
print "Version %s not found in the system, cannot load!" % ver
|
|
sys.exit(1)
|
|
|
|
# Remove any old docs for this version (still protected by a transaction while
|
|
# we perform the load)
|
|
curs.execute("DELETE FROM docs WHERE version=%(v)s", {'v': ver})
|
|
|
|
|
|
re_htmlfile = re.compile('[^/]*/doc/src/sgml/html/.*')
|
|
re_tarfile = re.compile('[^/]*/doc/postgres.tar.gz$')
|
|
for member in tf:
|
|
if re_htmlfile.match(member.name):
|
|
load_doc_file(os.path.basename(member.name), tf.extractfile(member))
|
|
if re_tarfile.match(member.name):
|
|
f = tf.extractfile(member)
|
|
inner_tar = tarfile.open(fileobj=f)
|
|
for inner_member in inner_tar:
|
|
if inner_member.name.endswith('.html') or inner_member.name.endswith('.htm'):
|
|
load_doc_file(inner_member.name, inner_tar.extractfile(inner_member))
|
|
tf.close()
|
|
|
|
transaction.commit_unless_managed()
|
|
|
|
if not quiet: print "Done (%i pages)." % pagecount
|
|
|