add parameter parser
This commit is contained in:
parent
7e484fa308
commit
cd6b03b0ff
@ -6,10 +6,11 @@ import requests, os, logging, re, json
|
||||
|
||||
class WPimport:
|
||||
# Constructor
|
||||
def __init__(self, basic, wordpress, logger):
|
||||
def __init__(self, basic, wordpress, logger, parser):
|
||||
self._basic = basic
|
||||
self._wordpress = wordpress
|
||||
self._logger = logger
|
||||
self._parser = parser
|
||||
|
||||
# Public method
|
||||
|
||||
@ -26,7 +27,7 @@ class WPimport:
|
||||
self._logger.info("Fichier en cours de traitement : {0}".format(file))
|
||||
with open(file, 'r') as f:
|
||||
content = f.read()
|
||||
soup = BeautifulSoup(content, 'html.parser')
|
||||
soup = BeautifulSoup(content, self._parser)
|
||||
articlebody = soup.find_all("div", class_="articlebody")
|
||||
if len(articlebody) > 0:
|
||||
self._addOrUpdatePost(soup)
|
||||
|
@ -10,6 +10,7 @@ if __name__ == '__main__':
|
||||
parser.add_argument("--debug", help="Verbosity", action="store_true")
|
||||
parser.add_argument("--logfile", help="Log file", default="")
|
||||
parser.add_argument("--quiet", help="No console output", action="store_true")
|
||||
parser.add_argument("--parser", help="Parser content", default="html.parser")
|
||||
|
||||
subparsers = parser.add_subparsers(dest="command")
|
||||
|
||||
@ -65,7 +66,7 @@ if __name__ == '__main__':
|
||||
exit(1)
|
||||
|
||||
basic = HTTPBasicAuth(args.user, password)
|
||||
importWp = WPImport.WPimport(basic, args.wordpress, logger)
|
||||
importWp = WPImport.WPimport(basic, args.wordpress, logger, args.parser)
|
||||
if len(args.file) > 0:
|
||||
importWp.fromFile(args.file.split(","))
|
||||
exit(0)
|
||||
|
Loading…
x
Reference in New Issue
Block a user