VladPolskiy/postgres-web
1
0
Fork 0
mirror of https://github.com/postgres/pgweb.git synced 2025-07-29 11:59:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add option to docload.py to make it run quietly

Browse Source 
Using commandline --quiet. To do this, switch to using the standard
python optionparser for commandline parsing.
This commit is contained in:
Magnus Hagander
2011-11-22 21:09:12 +01:00
parent 7e9742c678
commit bcce9e0aee
1 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
tools/docs/docload.py
View File

@ -7,6 +7,8 @@ import os
import tarfile import tarfile
import re import re
import tidy import tidy
from optparse import OptionParser
# Set up for accessing django # Set up for accessing django
from django.core.management import setup_environ from django.core.management import setup_environ
@ -17,6 +19,7 @@ setup_environ(settings)
from django.db import connection, transaction from django.db import connection, transaction
pagecount = 0 pagecount = 0
quiet = False
re_titlematch = re.compile('<title\s*>([^<]+)</title\s*>', re.IGNORECASE) re_titlematch = re.compile('<title\s*>([^<]+)</title\s*>', re.IGNORECASE)
@ -38,7 +41,7 @@ def load_doc_file(filename, f):
title = tm.group(1) title = tm.group(1)
else: else:
title = "" title = ""
print "--- file: %s (%s) ---" % (filename, title) if not quiet: print "--- file: %s (%s) ---" % (filename, title)
s = tidy.parseString(contents.encode('utf-8'), **tidyopts) 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)",{ curs.execute("INSERT INTO docs (file, version, title, content) VALUES (%(f)s, %(v)s, %(t)s, %(c)s)",{
@ -50,16 +53,18 @@ def load_doc_file(filename, f):
global pagecount global pagecount
pagecount += 1 pagecount += 1
## Your typical usage message
def Usage():
print "Usage: docload.py <version> <tarfile>"
sys.exit(1)
## Main execution ## Main execution
if len(sys.argv) != 3: parser = OptionParser(usage="usage: %prog [options] <version> <tarfile>")
Usage() 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] ver = sys.argv[1]
tarfilename = sys.argv[2] tarfilename = sys.argv[2]
@ -97,5 +102,5 @@ tf.close()
transaction.commit_unless_managed() transaction.commit_unless_managed()
print "Done (%i pages)." % pagecount if not quiet: print "Done (%i pages)." % pagecount