Add commandline parser

This commit is contained in:
Stefan Ritter 2012-01-21 00:42:23 +01:00
parent bdbc414fea
commit f473565b37
1 changed files with 31 additions and 0 deletions

View File

@ -24,6 +24,7 @@ from hashlib import md5
from glob import glob
from random import randint
from codecs import getwriter
from optparse import OptionParser
# print() will output ascii, but we want utf-8 (python3)
try:
@ -192,6 +193,36 @@ else:
else:
locale.setlocale(locale.LC_TIME, None)
# Commandline arguments
parser = OptionParser()
parser.add_option("-i", "--info", help="show statistics about your blog", action="store_true", dest="info")
parser.add_option("-t", "--tags", help="read all tags and create new index", action="store_true", dest="tags")
options, args = parser.parse_args()
if vars(options).values().count(True) > 1:
print("Too much arguments, just take one!")
sys.exit(0)
if options.info:
num_entries = len(glob(os.path.join(entries_dir, "*." + entries_suffix)))
num_comments = 0
comments = glob(os.path.join(entries_dir, "*.comments"))
for file in comments:
content = open(file, "r")
for line in content:
if line.startswith("-"):
num_comments += 1
content.close()
print("Number of entries: %s" % num_entries)
print("Number of comments: %s" % num_comments)
sys.exit(0)
if options.tags:
print("Tags are coming soon...")
sys.exit(0)
# Read POST variables
action = FieldStorage()
month_display = action.getvalue("m")