1
0
mirror of https://github.com/r4sas/py-i2phosts synced 2025-02-02 01:44:40 +00:00

injector: add command-line arguments parsing

And first option is: path to hosts.txt for parse
This commit is contained in:
Hidden Z 2010-10-13 17:32:34 +00:00
parent 40a708bb60
commit fd210c6fe3

View File

@ -2,6 +2,7 @@
import os import os
import sys import sys
import argparse
# django setup # django setup
DJANGO_SETTINGS_MODULE = 'settings' DJANGO_SETTINGS_MODULE = 'settings'
@ -10,10 +11,15 @@ sys.path.insert(1, DJANGO_PROJECT_PATH)
os.environ['DJANGO_SETTINGS_MODULE'] = DJANGO_SETTINGS_MODULE os.environ['DJANGO_SETTINGS_MODULE'] = DJANGO_SETTINGS_MODULE
from web.postkey.models import i2phost from web.postkey.models import i2phost
# hosts.txt we want to parse # parse command line options
hostsfile = os.environ['HOME'] + '/.i2p/hosts.txt' parser = argparse.ArgumentParser(
description='Hosts injector for py-i2phosts.',
epilog='Report bugs to http://zzz.i2p/topics/733')
parser.add_argument('-f', '--file', default=os.environ['HOME'] + '/.i2p/hosts.txt', dest='hostsfile',
help='hosts.txt for parsing')
args = parser.parse_args()
f = open(hostsfile, 'r') f = open(args.hostsfile, 'r')
for line in f: for line in f:
entry = line.split('=') entry = line.split('=')
# Check for already existed hosts in database to avoid adding duplicates. # Check for already existed hosts in database to avoid adding duplicates.