Compare commits
54 Commits
Author | SHA1 | Date | |
---|---|---|---|
b4d0fe8aa0 | |||
6401692d0d | |||
1fc9c48d2c | |||
d9c20cedcb | |||
9d41e57379 | |||
d88ae7ed44 | |||
50bf31d334 | |||
9b58b45ae8 | |||
418bea3778 | |||
5959ab5b2e | |||
e4eb1b6b68 | |||
9ed5ffe399 | |||
28b513e1b2 | |||
02f0c20bd0 | |||
1655217050 | |||
49c1552062 | |||
7f800c8f7b | |||
5399b12133 | |||
b7493206a2 | |||
8de7485775 | |||
eee14e08f1 | |||
cf2c1aa617 | |||
e17cace820 | |||
941776a7c1 | |||
72fbe0a364 | |||
1f8ea70b40 | |||
c6894648d1 | |||
d5ddf1601b | |||
aa8ac9cfcb | |||
82f9acd658 | |||
7593b68b6c | |||
5fe4b1f786 | |||
0445054dc8 | |||
be9ad9b934 | |||
e6328135da | |||
dded126926 | |||
bca529f3c3 | |||
5059a15826 | |||
20c4adb3cf | |||
74fa87ea73 | |||
03f833a2c3 | |||
9acb620f93 | |||
c6ccf98b1b | |||
95f5203727 | |||
9bbf769b40 | |||
f0b4ba5e27 | |||
47f504beb5 | |||
4b6b06aade | |||
d10867a983 | |||
6fba5f009a | |||
699cecad4f | |||
9f87f38347 | |||
55d62cebfb | |||
193b0e6ef7 |
1
.gitignore
vendored
1
.gitignore
vendored
@@ -2,3 +2,4 @@ backup*/
|
|||||||
wp-navigation
|
wp-navigation
|
||||||
*.log
|
*.log
|
||||||
__pycache__/
|
__pycache__/
|
||||||
|
wp-gallery
|
@@ -10,10 +10,23 @@ from lib.WPImport import WPimport
|
|||||||
from lib.WPExport import WPExport
|
from lib.WPExport import WPExport
|
||||||
from lib.WPRemove import WPRemove
|
from lib.WPRemove import WPRemove
|
||||||
from lib.WPChange import WPChange
|
from lib.WPChange import WPChange
|
||||||
|
from lib.WPMenu import WPMenu
|
||||||
|
|
||||||
def change(index, number, args, logger):
|
def errorRevert(logger, revert, tmp):
|
||||||
changeWp = WPChange(logger=logger, index_name=index, number_thread=number)
|
if revert is True:
|
||||||
changeWp.fromDirectory(args.directory)
|
files_tmp = glob.glob("{0}/*.json".format(tmp))
|
||||||
|
if len(files_tmp) == 0:
|
||||||
|
logger.error("Error revert, because files not found")
|
||||||
|
exit(1)
|
||||||
|
if len(files_tmp) != int(args.parallel):
|
||||||
|
for file_r in files_tmp:
|
||||||
|
os.remove(file_r)
|
||||||
|
logger.error("Error revert, because number files tmp is incompatible with parallel number")
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
def change(index, number, args, logger, tmp, revert):
|
||||||
|
changeWp = WPChange(logger=logger, index_name=index, number_thread=number, tmp=tmp)
|
||||||
|
changeWp.fromDirectory(args.directory, revert)
|
||||||
|
|
||||||
del changeWp
|
del changeWp
|
||||||
|
|
||||||
@@ -54,7 +67,7 @@ def download(name_thread, max_thread, url, logger, parser, directory, html, img,
|
|||||||
del exportWp
|
del exportWp
|
||||||
|
|
||||||
|
|
||||||
def importUrl(name_thread, max_thread, canalblog, logger, parser, wordpress, basic, serial, ssl_wordpress, ssl_canalblog, create, update, image, revert, tmp):
|
def importUrl(name_thread, max_thread, canalblog, logger, parser, wordpress, basic, serial, ssl_wordpress, ssl_canalblog, create, update, image, revert, tmp, author):
|
||||||
canalblog = canalblog.split(",")
|
canalblog = canalblog.split(",")
|
||||||
wordpress = wordpress.split(",")
|
wordpress = wordpress.split(",")
|
||||||
name = "Thread-{0}".format(int(name_thread) + 1)
|
name = "Thread-{0}".format(int(name_thread) + 1)
|
||||||
@@ -75,7 +88,7 @@ def importUrl(name_thread, max_thread, canalblog, logger, parser, wordpress, bas
|
|||||||
exportWp.getUrlPage(name_thread, max_thread)
|
exportWp.getUrlPage(name_thread, max_thread)
|
||||||
del exportWp
|
del exportWp
|
||||||
for j in wordpress:
|
for j in wordpress:
|
||||||
importWp = WPimport(name=name, basic=basic, wordpress=j, logger=logger, parser=parser, ssl_wordpress=ssl_wordpress, no_create=create, no_update=update, no_image=image, tmp=tmp)
|
importWp = WPimport(name=name, basic=basic, wordpress=j, logger=logger, parser=parser, ssl_wordpress=ssl_wordpress, no_create=create, no_update=update, no_image=image, tmp=tmp, author=author)
|
||||||
for k in ["article", "page"]:
|
for k in ["article", "page"]:
|
||||||
for l in ["publications", "principal"]:
|
for l in ["publications", "principal"]:
|
||||||
importWp.fromUrl(l, k)
|
importWp.fromUrl(l, k)
|
||||||
@@ -97,7 +110,7 @@ def importUrl(name_thread, max_thread, canalblog, logger, parser, wordpress, bas
|
|||||||
if not revert:
|
if not revert:
|
||||||
exportWp.getUrlPage(name_thread, max_thread)
|
exportWp.getUrlPage(name_thread, max_thread)
|
||||||
del exportWp
|
del exportWp
|
||||||
importWp = WPimport(name=name, basic=basic, wordpress=wordpress[i], logger=logger, parser=parser, ssl_wordpress=ssl_wordpress, no_create=create, no_update=update, no_image=image, tmp=tmp)
|
importWp = WPimport(name=name, basic=basic, wordpress=wordpress[i], logger=logger, parser=parser, ssl_wordpress=ssl_wordpress, no_create=create, no_update=update, no_image=image, tmp=tmp, author=author)
|
||||||
|
|
||||||
for k in ["article", "page"]:
|
for k in ["article", "page"]:
|
||||||
for l in ["publications", "principal"]:
|
for l in ["publications", "principal"]:
|
||||||
@@ -106,15 +119,15 @@ def importUrl(name_thread, max_thread, canalblog, logger, parser, wordpress, bas
|
|||||||
del importWp
|
del importWp
|
||||||
|
|
||||||
|
|
||||||
def importDirectory(name_thread, max_thread, directory, logger, parser, wordpress, basic, serial, ssl_wordpress, create, update, image):
|
def importDirectory(name_thread, max_thread, directory, logger, parser, wordpress, basic, serial, ssl_wordpress, create, update, image, revert, author):
|
||||||
name = "Thread-{0}".format(int(name_thread) + 1)
|
name = "Thread-{0}".format(int(name_thread) + 1)
|
||||||
directory = directory.split(",")
|
directory = directory.split(",")
|
||||||
wordpress = wordpress.split(",")
|
wordpress = wordpress.split(",")
|
||||||
if serial is False:
|
if serial is False:
|
||||||
for i in wordpress:
|
for i in wordpress:
|
||||||
importWp = WPimport(name=name, basic=basic, wordpress=i, logger=logger, parser=parser, ssl_wordpress=ssl_wordpress, no_create=create, no_update=update, no_image=image)
|
importWp = WPimport(name=name, basic=basic, wordpress=i, logger=logger, parser=parser, ssl_wordpress=ssl_wordpress, no_create=create, no_update=update, no_image=image, author=author)
|
||||||
for j in directory:
|
for j in directory:
|
||||||
importWp.fromDirectory(j, name_thread, max_thread)
|
importWp.fromDirectory(j, name_thread, max_thread, revert)
|
||||||
del importWp
|
del importWp
|
||||||
|
|
||||||
else:
|
else:
|
||||||
@@ -122,8 +135,8 @@ def importDirectory(name_thread, max_thread, directory, logger, parser, wordpres
|
|||||||
logger.error("{0} : Error : Number directory is different than wordpress".format(name))
|
logger.error("{0} : Error : Number directory is different than wordpress".format(name))
|
||||||
exit(1)
|
exit(1)
|
||||||
for i in range(0, len(wordpress)-1):
|
for i in range(0, len(wordpress)-1):
|
||||||
importWp = WPimport(name=name, basic=basic, wordpress=wordpress[i], logger=logger, parser=parser, ssl_wordpress=ssl_wordpress, no_create=create, no_update=update, no_image=image)
|
importWp = WPimport(name=name, basic=basic, wordpress=wordpress[i], logger=logger, parser=parser, ssl_wordpress=ssl_wordpress, no_create=create, no_update=update, no_image=image, author=author)
|
||||||
importWp.fromDirectory(directory[i])
|
importWp.fromDirectory(directory[i], name_thread, max_thread, revert)
|
||||||
del importWp
|
del importWp
|
||||||
|
|
||||||
|
|
||||||
@@ -157,6 +170,7 @@ if __name__ == '__main__':
|
|||||||
import_parser.add_argument("--no-create", help="No create post", dest="create", default="store_false", action="store_true")
|
import_parser.add_argument("--no-create", help="No create post", dest="create", default="store_false", action="store_true")
|
||||||
import_parser.add_argument("--no-update", help="No update post", dest="update", default="store_false", action="store_true")
|
import_parser.add_argument("--no-update", help="No update post", dest="update", default="store_false", action="store_true")
|
||||||
import_parser.add_argument("--no-image", help="No image add or update", dest="image", default="store_false", action="store_true")
|
import_parser.add_argument("--no-image", help="No image add or update", dest="image", default="store_false", action="store_true")
|
||||||
|
import_parser.add_argument("--author-album", dest=author, help="Define author for page album", default="")
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@@ -191,6 +205,12 @@ if __name__ == '__main__':
|
|||||||
default="",
|
default="",
|
||||||
help="File")
|
help="File")
|
||||||
|
|
||||||
|
menu_parser = subparsers.add_parser("menu")
|
||||||
|
menu_parser.add_argument("--user", help="wordpress user", required=True)
|
||||||
|
menu_parser.add_argument("--password", help="password wordpress's user", default="")
|
||||||
|
menu_parser.add_argument("--file", help="HTML file", default="")
|
||||||
|
menu_parser.add_argument("--canalblog", help="URL Canalblog", default="")
|
||||||
|
menu_parser.add_argument("--wordpress", help="URL Wordpress", required=True)
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
@@ -228,7 +248,7 @@ if __name__ == '__main__':
|
|||||||
|
|
||||||
os.makedirs(args.tmp, exist_ok=True)
|
os.makedirs(args.tmp, exist_ok=True)
|
||||||
|
|
||||||
if args.command == "import" or args.command == "remove":
|
if args.command == "import" or args.command == "remove" or args.command == "menu":
|
||||||
password = args.password
|
password = args.password
|
||||||
if len(args.password) == 0:
|
if len(args.password) == 0:
|
||||||
password = getpass()
|
password = getpass()
|
||||||
@@ -239,20 +259,26 @@ if __name__ == '__main__':
|
|||||||
basic = HTTPBasicAuth(args.user, password)
|
basic = HTTPBasicAuth(args.user, password)
|
||||||
if args.command == "import":
|
if args.command == "import":
|
||||||
wordpress = args.wordpress.split(",")
|
wordpress = args.wordpress.split(",")
|
||||||
importWp = WPimport(basic=basic, wordpress="", logger=logger, parser=args.parser, ssl_wordpress=ssl_wordpress)
|
importWp = WPimport(basic=basic, wordpress="", logger=logger, parser=args.parser, ssl_wordpress=ssl_wordpress, author=args.author)
|
||||||
if len(args.file) > 0:
|
if len(args.file) > 0:
|
||||||
for i in wordpress:
|
for i in wordpress:
|
||||||
importWp.setUrl(i)
|
importWp.setUrl(i)
|
||||||
importWp.fromFile(files=args.file.split(","))
|
importWp.fromFile(files=args.file.split(","))
|
||||||
|
menuWp = WPMenu(name="Thread-1", basic=basic, wordpress=args.wordpress, logger=logger, parser=args.parser, ssl_canalblog=ssl_canalblog, ssl_wordpress=ssl_wordpress)
|
||||||
|
menuWp.fromFile("{0}".format(args.file.split(",")[0]))
|
||||||
if len(args.directory) > 0:
|
if len(args.directory) > 0:
|
||||||
try:
|
try:
|
||||||
with futures.ThreadPoolExecutor(max_workers=int(args.parallel)) as ex:
|
with futures.ThreadPoolExecutor(max_workers=int(args.parallel)) as ex:
|
||||||
wait_for = [ ex.submit(remove, i, args.parallel, args, basic, logger, ssl_wordpress) for i in range(0, int(args.parallel)) ]
|
wait_for = [ ex.submit(remove, i, args.parallel, args, basic, logger, ssl_wordpress) for i in range(0, int(args.parallel)) ]
|
||||||
wait(wait_for, return_when=ALL_COMPLETED)
|
wait(wait_for, return_when=ALL_COMPLETED)
|
||||||
|
errorRevert(logger, args.revert, args.tmp)
|
||||||
wait_for = [
|
wait_for = [
|
||||||
ex.submit(importDirectory, i, int(args.parallel), args.directory, logger, args.parser, args.wordpress, basic, args.serial, ssl_wordpress, args.create, args.update, args.image)
|
ex.submit(importDirectory, i, int(args.parallel), args.directory, logger, args.parser, args.wordpress, basic, args.serial, ssl_wordpress, args.create, args.update, args.image, args.revert, args.author)
|
||||||
for i in range(0, int(args.parallel))
|
for i in range(0, int(args.parallel))
|
||||||
]
|
]
|
||||||
|
wait(wait_for, return_when=ALL_COMPLETED)
|
||||||
|
menuWp = WPMenu(name="Thread-1", basic=basic, wordpress=args.wordpress, logger=logger, parser=args.parser, ssl_canalblog=ssl_canalblog, ssl_wordpress=ssl_wordpress)
|
||||||
|
menuWp.fromFile("{0}/index.html".format(args.directory))
|
||||||
except Exception as err:
|
except Exception as err:
|
||||||
logger.error("Threading error : {0}".format(err))
|
logger.error("Threading error : {0}".format(err))
|
||||||
if len(args.canalblog) > 0:
|
if len(args.canalblog) > 0:
|
||||||
@@ -260,19 +286,15 @@ if __name__ == '__main__':
|
|||||||
with futures.ThreadPoolExecutor(max_workers=int(args.parallel)) as ex:
|
with futures.ThreadPoolExecutor(max_workers=int(args.parallel)) as ex:
|
||||||
wait_for = [ ex.submit(remove, i, args.parallel, args, basic, logger, ssl_wordpress) for i in range(0, int(args.parallel)) ]
|
wait_for = [ ex.submit(remove, i, args.parallel, args, basic, logger, ssl_wordpress) for i in range(0, int(args.parallel)) ]
|
||||||
wait(wait_for, return_when=ALL_COMPLETED)
|
wait(wait_for, return_when=ALL_COMPLETED)
|
||||||
if args.revert is True:
|
errorRevert(logger, args.revert, args.tmp)
|
||||||
files_tmp = glob.glob("{0}/*.json".format(args.tmp))
|
|
||||||
if len(files_tmp) == 0:
|
|
||||||
logger.error("Error revert, because files not found")
|
|
||||||
exit(1)
|
|
||||||
if len(files_tmp) != int(args.parallel):
|
|
||||||
for file_r in files_tmp:
|
|
||||||
os.remove(file_r)
|
|
||||||
|
|
||||||
wait_for = [
|
wait_for = [
|
||||||
ex.submit(importUrl, i, int(args.parallel), args.canalblog, logger, args.parser, args.wordpress, basic, args.serial, ssl_wordpress, ssl_canalblog, args.create, args.update, args.image, args.revert, args.tmp)
|
ex.submit(importUrl, i, int(args.parallel), args.canalblog, logger, args.parser, args.wordpress, basic, args.serial, ssl_wordpress, ssl_canalblog, args.create, args.update, args.image, args.revert, args.tmp, args.author)
|
||||||
for i in range(0, int(args.parallel))
|
for i in range(0, int(args.parallel))
|
||||||
]
|
]
|
||||||
|
wait(wait_for, return_when=ALL_COMPLETED)
|
||||||
|
menuWp = WPMenu(name="Thread-1", basic=basic, wordpress=args.wordpress, logger=logger, parser=args.parser, ssl_canalblog=ssl_canalblog, ssl_wordpress=ssl_wordpress)
|
||||||
|
menuWp.fromUrl(args.canalblog)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
except Exception as err:
|
except Exception as err:
|
||||||
@@ -328,7 +350,8 @@ if __name__ == '__main__':
|
|||||||
if len(args.directory) > 0:
|
if len(args.directory) > 0:
|
||||||
try:
|
try:
|
||||||
with futures.ThreadPoolExecutor(max_workers=int(args.parallel)) as ex:
|
with futures.ThreadPoolExecutor(max_workers=int(args.parallel)) as ex:
|
||||||
wait_for = [ ex.submit(change, i, args.parallel, args, logger) for i in range(0, int(args.parallel)) ]
|
errorRevert(logger, args.revert, args.tmp)
|
||||||
|
wait_for = [ ex.submit(change, i, args.parallel, args, logger, args.tmp, args.revert) for i in range(0, int(args.parallel)) ]
|
||||||
except Exception as err:
|
except Exception as err:
|
||||||
logger.error("Thread error for remove : {0}".format(err))
|
logger.error("Thread error for remove : {0}".format(err))
|
||||||
if len(args.file) > 0:
|
if len(args.file) > 0:
|
||||||
@@ -336,3 +359,11 @@ if __name__ == '__main__':
|
|||||||
for filei in args.file.split(","):
|
for filei in args.file.split(","):
|
||||||
changeWp.fromFile(filei)
|
changeWp.fromFile(filei)
|
||||||
exit(0)
|
exit(0)
|
||||||
|
|
||||||
|
if args.command == "menu":
|
||||||
|
menuWp = WPMenu(name="Thread-1", basic=basic, wordpress=args.wordpress, logger=logger, parser=args.parser, ssl_canalblog=ssl_canalblog, ssl_wordpress=ssl_wordpress)
|
||||||
|
if len(args.file) > 0:
|
||||||
|
menuWp.fromFile(args.file)
|
||||||
|
if len(args.canalblog) > 0:
|
||||||
|
menuWp.fromUrl(args.canalblog)
|
||||||
|
exit(0)
|
@@ -4,11 +4,13 @@ import requests, os, logging, re, json
|
|||||||
|
|
||||||
class WPChange:
|
class WPChange:
|
||||||
# Constructor
|
# Constructor
|
||||||
def __init__(self, index_name=1, number_thread=1, logger=None, parser="html.parser"):
|
def __init__(self, index_name=1, number_thread=1, logger=None, parser="html.parser", tmp="/tmp/import_export_canablog"):
|
||||||
self._name = "Thread-{0}".format(index_name)
|
self._name = "Thread-{0}".format(index_name)
|
||||||
self._logger = logger
|
self._logger = logger
|
||||||
self._number_thread = number_thread
|
self._number_thread = number_thread
|
||||||
self._parser = parser
|
self._parser = parser
|
||||||
|
self._tmp = tmp
|
||||||
|
self._index_name = index_name
|
||||||
|
|
||||||
# Destructor
|
# Destructor
|
||||||
def __del__(self):
|
def __del__(self):
|
||||||
@@ -38,18 +40,61 @@ class WPChange:
|
|||||||
|
|
||||||
## From directory
|
## From directory
|
||||||
|
|
||||||
def fromDirectory(self, directory="", number_thread=1, max_thread=1):
|
|
||||||
|
def fromDirectory(self, directory="", revert=False):
|
||||||
|
self._directory = directory
|
||||||
directory = "{0}/archives".format(directory)
|
directory = "{0}/archives".format(directory)
|
||||||
directories = self._getDirectories([], "{0}".format(directory))
|
directories = self._getDirectories([], "{0}".format(directory))
|
||||||
if len(directories) > 0:
|
if len(directories) > 0:
|
||||||
files = self._getFiles(directories)
|
files = self._getFiles(directories)
|
||||||
self.fromFile(files, number_thread, max_thread)
|
if revert is False:
|
||||||
|
self._tmpFiles(files=files, number_thread=self._index_name, max_thread=self._number_thread)
|
||||||
|
self._fromFileTmp()
|
||||||
else:
|
else:
|
||||||
self._logger.error("{0} : No files for {1}".format(self._name, directory))
|
self._logger.error("{0} : No files for {1}".format(self._name, directory))
|
||||||
|
|
||||||
|
|
||||||
|
def fromFile(self, files=[]):
|
||||||
|
for i in range(0, len(files)):
|
||||||
|
if os.path.exists(files[i]):
|
||||||
|
self._logger.info("{0} : ({1}/{2}) File is being processed : {3}".format(self._name, i+1, len(files), files[i]))
|
||||||
|
self._change(files[i])
|
||||||
|
|
||||||
|
|
||||||
# Private method
|
# Private method
|
||||||
|
|
||||||
|
def _fromFileTmp(self):
|
||||||
|
try:
|
||||||
|
with open("{0}/{1}.json".format(self._tmp, self._name)) as file:
|
||||||
|
files = json.loads(file.read())
|
||||||
|
self._logger.debug("{0} : size of webpage : {1}".format(self._name, len(files)))
|
||||||
|
for i in range(0, len(files)):
|
||||||
|
if os.path.exists(files[i]):
|
||||||
|
self._logger.info("{0} : ({1}/{2}) File is being processed : {3}".format(self._name, i+1, len(files), files[i]))
|
||||||
|
self._change(files[i])
|
||||||
|
except Exception as ex:
|
||||||
|
self._logger.error("{0} : Read file json from tmp : {1}".format(self._name, ex))
|
||||||
|
|
||||||
|
|
||||||
|
def _tmpFiles(self, files=[], number_thread=1, max_thread=1):
|
||||||
|
print()
|
||||||
|
divFiles = int(len(files) / int(max_thread))
|
||||||
|
currentRangeFiles = int(divFiles * (int(number_thread)+1))
|
||||||
|
firstRange = int(currentRangeFiles - divFiles)
|
||||||
|
self._logger.debug("{0} : index : {1}".format(self._name,number_thread))
|
||||||
|
|
||||||
|
self._logger.debug("{0} : first range : {1}".format(self._name,firstRange))
|
||||||
|
self._logger.debug("{0} : last range : {1}".format(self._name,currentRangeFiles))
|
||||||
|
webpage = []
|
||||||
|
for i in range(firstRange, currentRangeFiles):
|
||||||
|
webpage.append(files[i])
|
||||||
|
|
||||||
|
try:
|
||||||
|
string_webpage = json.dumps(webpage)
|
||||||
|
open("{0}/{1}.json".format(self._tmp, self._name), "wt").write(string_webpage)
|
||||||
|
except Exception as ex:
|
||||||
|
self._logger.error("{0} : Error for writing webpage : {1}".format(self._name, ex))
|
||||||
|
|
||||||
## Get all files
|
## Get all files
|
||||||
|
|
||||||
def _getFiles(self, item):
|
def _getFiles(self, item):
|
||||||
|
275
lib/WPImport.py
275
lib/WPImport.py
@@ -8,7 +8,7 @@ from requests.packages.urllib3.util.retry import Retry
|
|||||||
|
|
||||||
class WPimport:
|
class WPimport:
|
||||||
# Constructor
|
# Constructor
|
||||||
def __init__(self, name="Thread-0", basic=None, wordpress="", logger=None, parser="html.parser", ssl_wordpress=True, no_create=False, no_update=False, no_image=False, tmp="/tmp/import_export_canablog"):
|
def __init__(self, name="Thread-0", basic=None, wordpress="", logger=None, parser="html.parser", ssl_wordpress=True, no_create=False, no_update=False, no_image=False, tmp="/tmp/import_export_canablog", author=""):
|
||||||
self._name = name
|
self._name = name
|
||||||
self._basic = basic
|
self._basic = basic
|
||||||
self._wordpress = wordpress
|
self._wordpress = wordpress
|
||||||
@@ -29,6 +29,7 @@ class WPimport:
|
|||||||
self._no_update = no_update
|
self._no_update = no_update
|
||||||
self._no_image = no_image
|
self._no_image = no_image
|
||||||
self._tmp = tmp
|
self._tmp = tmp
|
||||||
|
self._author = author
|
||||||
|
|
||||||
# Destructor
|
# Destructor
|
||||||
def __del__(self):
|
def __del__(self):
|
||||||
@@ -54,6 +55,10 @@ class WPimport:
|
|||||||
articlebody = soup.find_all("div", class_="articlebody")
|
articlebody = soup.find_all("div", class_="articlebody")
|
||||||
if len(articlebody) > 0:
|
if len(articlebody) > 0:
|
||||||
self._addOrUpdatePost(soup)
|
self._addOrUpdatePost(soup)
|
||||||
|
else:
|
||||||
|
albumbody = soup.find_all("div", class_="albumbody")
|
||||||
|
if len(albumbody) > 0:
|
||||||
|
self._addOrUpdateAlbum(soup)
|
||||||
else:
|
else:
|
||||||
self._addOrUpdateFeaturedMedia(soup)
|
self._addOrUpdateFeaturedMedia(soup)
|
||||||
del webpage_content[first][second][i]
|
del webpage_content[first][second][i]
|
||||||
@@ -74,33 +79,226 @@ class WPimport:
|
|||||||
self._logger.error("{0} : Read file json from tmp : {1}".format(self._name, ex))
|
self._logger.error("{0} : Read file json from tmp : {1}".format(self._name, ex))
|
||||||
|
|
||||||
|
|
||||||
def fromDirectory(self, directory="", number_thread=1, max_thread=1):
|
def fromDirectory(self, directory="", number_thread=1, max_thread=1, revert=False):
|
||||||
self._directory = directory
|
self._directory = directory
|
||||||
directory = "{0}/archives".format(directory)
|
directory = "{0}/archives".format(directory)
|
||||||
directories = self._getDirectories([], "{0}".format(directory))
|
directories = self._getDirectories([], "{0}".format(directory))
|
||||||
if len(directories) > 0:
|
if len(directories) > 0:
|
||||||
files = self._getFiles(directories)
|
files = self._getFiles(directories)
|
||||||
self.fromFile(files=files, number_thread=number_thread, max_thread=max_thread)
|
if revert is False:
|
||||||
|
self._tmpFiles(files=files, number_thread=number_thread, max_thread=max_thread)
|
||||||
|
self._fromFileTmp()
|
||||||
else:
|
else:
|
||||||
self._logger.error("{0} : No files for {1}".format(self._name, directory))
|
self._logger.error("{0} : No files for {1}".format(self._name, directory))
|
||||||
|
|
||||||
|
|
||||||
|
def fromFile(self, files=[]):
|
||||||
|
for i in range(0, len(files)):
|
||||||
|
|
||||||
|
|
||||||
def fromFile(self, files=[], number_thread=1, max_thread=1):
|
|
||||||
divFiles = int(len(files) / max_thread)
|
|
||||||
currentRangeFiles = int(divFiles * (number_thread+1))
|
|
||||||
firstRange = int(currentRangeFiles - divFiles)
|
|
||||||
self._logger.debug("{0} : index : {1}".format(self._name,number_thread))
|
|
||||||
|
|
||||||
self._logger.debug("{0} : first range : {1}".format(self._name,firstRange))
|
|
||||||
self._logger.debug("{0} : last range : {1}".format(self._name,currentRangeFiles))
|
|
||||||
|
|
||||||
for i in range(firstRange, currentRangeFiles):
|
|
||||||
if os.path.exists(files[i]):
|
if os.path.exists(files[i]):
|
||||||
self._logger.info("{0} : ({1}/{2}) File is being processed : {3}".format(self._name, i+1, currentRangeFiles + 1, files[i]))
|
self._logger.info("{0} : ({1}/{2}) File is being processed : {3}".format(self._name, i+1, len(files), files[i]))
|
||||||
|
with open(files[i], 'r') as f:
|
||||||
|
content = f.read()
|
||||||
|
self._logger.debug("{0} : Size of article : {1}".format(self._name, len(content)))
|
||||||
|
soup = BeautifulSoup(content, self._parser)
|
||||||
|
articlebody = soup.find_all("div", class_="articlebody")
|
||||||
|
self._logger.debug("{0} : Number of article : {1}".format(self._name, len(articlebody)))
|
||||||
|
if len(articlebody) > 0:
|
||||||
|
self._addOrUpdatePost(soup)
|
||||||
|
else:
|
||||||
|
albumbody = soup.find_all("div", class_="albumbody")
|
||||||
|
if len(albumbody) > 0:
|
||||||
|
self._addOrUpdateAlbum(soup)
|
||||||
|
else:
|
||||||
|
self._addOrUpdateFeaturedMedia(soup)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# Private method
|
||||||
|
|
||||||
|
|
||||||
|
def _getAuthor(self, author):
|
||||||
|
params = {"search":author, "per_page":100}
|
||||||
|
author = 0
|
||||||
|
try:
|
||||||
|
self._logger.info("{0} : Search author : {1}".format(self._name, author))
|
||||||
|
page = self._request.get("{1}://{0}/wp-json/wp/v2/users".format(self._wordpress, self._protocol), auth=self._basic, headers=self._headers_json, params=params)
|
||||||
|
self._logger.debug("{0} : End Search author : {1}".format(self._name, author))
|
||||||
|
self._logger.debug("{0} : Debug requests : {1}".format(self._name, page.content))
|
||||||
|
if page.status_code == 200:
|
||||||
|
self._logger.info("{0} : Get author : {1}".format(self._name, author))
|
||||||
|
result = page.json()
|
||||||
|
for a in result:
|
||||||
|
author = a["id"]
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : Connection error with status code for get author : {1}".format(self._name, page.status_code))
|
||||||
|
self._logger.debug("{0} : {1}".format(page.content))
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for get author : {1}".format(self._name, err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for get author : {1}".format(self._name, err))
|
||||||
|
return author
|
||||||
|
|
||||||
|
def _addOrUpdateAlbum(self, soup):
|
||||||
|
self._logger.info("{0} : Add/Update Album".format(self._name))
|
||||||
|
albumbody = soup.find("div", class_="albumbody")
|
||||||
|
albumtitle = albumbody.find("h2").get_text()
|
||||||
|
self._logger.debug("{0} : Title of the album : {1}".format(self._name, albumtitle))
|
||||||
|
albumdesc = albumbody.find("div", class_="albumdesc").find("p")
|
||||||
|
img_a = albumbody.find_all("img")
|
||||||
|
list_img = []
|
||||||
|
page_is_exist = False
|
||||||
|
|
||||||
|
if self._no_image is False:
|
||||||
|
self._logger.debug("{0} : Number of image's tag : {1}".format(self._name, len(img_a)))
|
||||||
|
|
||||||
|
for i in img_a:
|
||||||
|
new_img = {}
|
||||||
|
href_img = i.get("src")
|
||||||
|
href_img_o = urlparse(href_img)
|
||||||
|
try:
|
||||||
|
if len(href_img_o.netloc) > 0:
|
||||||
|
img_ok = False
|
||||||
|
href_img = href_img.replace("_q", "_o")
|
||||||
|
page_img = self._request.get(href_img)
|
||||||
|
if page_img.status_code == 200:
|
||||||
|
img_ok = True
|
||||||
|
else:
|
||||||
|
if os.path.exists("{0}/..{1}".format(self._directory, href_img)):
|
||||||
|
page_img = open("{0}/..{1}".format(self._directory, href_img), "r")
|
||||||
|
img_ok = True
|
||||||
|
self._logger.debug("{0} : Status code for image {1} : {2}".format(self._name, href_img, page_img.status_code))
|
||||||
|
if img_ok is True:
|
||||||
|
media=self._addOrUpdateMedia(href_img, page_img)
|
||||||
|
new_img["id"]=media["id"]
|
||||||
|
new_img["new_src"]=media["rendered"]
|
||||||
|
list_img.append(new_img)
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for get image : {1}".format(self._name, err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for get image : {1}".format(self._name, err))
|
||||||
|
exit(1)
|
||||||
|
self._logger.debug("{0} content img : {1}".format(self._name, list_img))
|
||||||
|
content_html = ""
|
||||||
|
if len(list_img) > 0:
|
||||||
|
|
||||||
|
content_html = "<!-- wp:paragraph -->\n{0}\n<!-- /wp:paragraph -->\n\n".format(albumdesc)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
for i in range(0, len(list_img)):
|
||||||
|
content_html = content_html + "<!-- wp:image {\"id\":id-image,\"sizeSlug\":\"large\",\"linkDestination\":\"none\"} --><figure class=\"wp-block-image size-large\"><img src=\"src-image\" alt=\"\" class=\"wp-image-id-image\"/></figure><!-- /wp:image -->\n\n".replace("id-image", str(list_img[i]["id"])).replace("src-image", list_img[i]["new_src"])
|
||||||
|
|
||||||
|
self._logger.info("{0} : content html : {1}".format(self._name, content_html))
|
||||||
|
if len(content_html) > 0:
|
||||||
|
data = {"title":albumtitle, "content":content_html, "status":"publish"}
|
||||||
|
if len(self._author) > 0:
|
||||||
|
author = self._getAuthor(self._author)
|
||||||
|
if author != 0:
|
||||||
|
data = {"title":albumtitle, "content":content_html, "status":"publish", "author":author}
|
||||||
|
self._logger.debug("{0} : data for album page : {1}".format(self._name, data))
|
||||||
|
for index in range(1,10):
|
||||||
|
params = {"search": albumtitle, "per_page":100, "page": index}
|
||||||
|
try:
|
||||||
|
self._logger.info("{0} : Search post with index {2} : {1}".format(self._name, albumtitle, index))
|
||||||
|
page = self._request.get("{1}://{0}/wp-json/wp/v2/pages".format(self._wordpress, self._protocol), auth=self._basic, params=params, headers=self._headers_json)
|
||||||
|
if page.status_code == 200:
|
||||||
|
self._logger.debug("{0} : Encoding : {1}".format(self._name, page.encoding))
|
||||||
|
page.encoding = "utf-8"
|
||||||
|
result = page.json()
|
||||||
|
if len(result) == 0:
|
||||||
|
break
|
||||||
|
self._logger.info("{0} : Number result posts : {1}".format(self._name, len(result)))
|
||||||
|
count = 0
|
||||||
|
for i in result:
|
||||||
|
title_rendered = i["title"]["rendered"]
|
||||||
|
self._logger.info("{0} : Search title pages for |{2}| : |{1}|".format(self._name, title_rendered, albumtitle))
|
||||||
|
if len(title_rendered) != len(albumtitle):
|
||||||
|
title_rendered = self._replaceCaracter(title_rendered)
|
||||||
|
self._logger.debug("{0} : Search title pages for |{2}| : |{1}|".format(self._name, title_rendered, albumtitle))
|
||||||
|
self._logger.debug("{0} : SIze of title : {1} - {2}".format(self._name, len(albumtitle), len(title_rendered)))
|
||||||
|
if title_rendered == albumtitle:
|
||||||
|
if self._no_update is False:
|
||||||
|
page_is_exist = True
|
||||||
|
post_id = i["id"]
|
||||||
|
count = count + 1
|
||||||
|
if count > 1:
|
||||||
|
self._logger.info("{0} : Page {1} is double and going to delete".format(self._name, albumtitle))
|
||||||
|
try:
|
||||||
|
params = {"force":1}
|
||||||
|
page = self._request.delete("{2}://{0}/wp-json/wp/v2/pages/{1}".format(self._wordpress, post_id, self._protocol), auth=self._basic, headers=self._headers_json, params=params)
|
||||||
|
if page.status_code == 200:
|
||||||
|
self._logger.info("{0} : Page deleted : {1}".format(self._name, albumtitle))
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : Page not updated due status code : {1}".format(self._name, page.status_code))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, page.content))
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for deleted page : {1}".format(self._name, err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for deleted page : {1}".format(self._name, err))
|
||||||
|
|
||||||
|
else:
|
||||||
|
self._logger.debug("{0} : Data for page to update : {1}".format(self._name, i))
|
||||||
|
self._logger.info("{0} : Page {1} already exist and going to update".format(self._name, albumtitle))
|
||||||
|
|
||||||
|
try:
|
||||||
|
page = self._request.post("{2}://{0}/wp-json/wp/v2/pages/{1}".format(self._wordpress, post_id, self._protocol), auth=self._basic, headers=self._headers_json, data=json.dumps(data))
|
||||||
|
|
||||||
|
if page.status_code == 200:
|
||||||
|
result = page.json()
|
||||||
|
self._logger.info("{0} : page updated : {1}".format(self._name, albumtitle))
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : page not updated due status code : {1}".format(self._name, page.status_code))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, page.content))
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for update page : {1}".format(self._name, err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for update page : {1}".format(self._name, err))
|
||||||
|
if page.status_code == 400:
|
||||||
|
self._logger.error("{0} : Connection for update post unauthorized : {1}".format(self._name, page.status_code))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, page.content))
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : Connection for update page error with status code : {1}".format(self._name, page.status_code))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, page.content))
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for search page : {1}".format(self._name, err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for search page : {1}".format(self._name, err))
|
||||||
|
|
||||||
|
if page_is_exist is False and self._no_create is False:
|
||||||
|
try:
|
||||||
|
self._logger.info("{0} : Creating page : {1}".format(self._name, data["title"]))
|
||||||
|
page = self._request.post("{1}://{0}/wp-json/wp/v2/pages".format(self._wordpress, self._protocol), auth=self._basic, headers=self._headers_json, data=json.dumps(data))
|
||||||
|
|
||||||
|
if page.status_code == 201:
|
||||||
|
result = page.json()
|
||||||
|
self._logger.info("{0} : page added : {1}".format(self._name, result["title"]["raw"]))
|
||||||
|
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : page not added due status code : {1}".format(self._name, r.status_code))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, r.content))
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for create page : {1}".format(self._name, err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for create page : {1}".format(self._name, err))
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def _fromFileTmp(self):
|
||||||
|
try:
|
||||||
|
with open("{0}/{1}.json".format(self._tmp, self._name)) as file:
|
||||||
|
files = json.loads(file.read())
|
||||||
|
self._logger.debug("{0} : size of webpage : {1}".format(self._name, len(files)))
|
||||||
|
for i in range(0, len(files)):
|
||||||
|
if os.path.exists(files[i]):
|
||||||
|
self._logger.info("{0} : ({1}/{2}) File is being processed : {3}".format(self._name, i+1, len(files), files[i]))
|
||||||
with open(files[i], 'r') as f:
|
with open(files[i], 'r') as f:
|
||||||
content = f.read()
|
content = f.read()
|
||||||
self._logger.debug("{0} : Size of article : {1}".format(self._name, len(content)))
|
self._logger.debug("{0} : Size of article : {1}".format(self._name, len(content)))
|
||||||
@@ -111,9 +309,28 @@ class WPimport:
|
|||||||
self._addOrUpdatePost(soup)
|
self._addOrUpdatePost(soup)
|
||||||
else:
|
else:
|
||||||
self._addOrUpdateFeaturedMedia(soup)
|
self._addOrUpdateFeaturedMedia(soup)
|
||||||
|
except Exception as ex:
|
||||||
|
self._logger.error("{0} : Read file json from tmp : {1}".format(self._name, ex))
|
||||||
|
|
||||||
|
|
||||||
# Private method
|
def _tmpFiles(self, files=[], number_thread=1, max_thread=1):
|
||||||
|
divFiles = int(len(files) / max_thread)
|
||||||
|
currentRangeFiles = int(divFiles * (number_thread+1))
|
||||||
|
firstRange = int(currentRangeFiles - divFiles)
|
||||||
|
self._logger.debug("{0} : index : {1}".format(self._name,number_thread))
|
||||||
|
|
||||||
|
self._logger.debug("{0} : first range : {1}".format(self._name,firstRange))
|
||||||
|
self._logger.debug("{0} : last range : {1}".format(self._name,currentRangeFiles))
|
||||||
|
webpage = []
|
||||||
|
for i in range(firstRange, currentRangeFiles):
|
||||||
|
webpage.append(files[i])
|
||||||
|
|
||||||
|
try:
|
||||||
|
string_webpage = json.dumps(webpage)
|
||||||
|
open("{0}/{1}.json".format(self._tmp, self._name), "wt").write(string_webpage)
|
||||||
|
except Exception as ex:
|
||||||
|
self._logger.error("{0} : Error for writing webpage : {1}".format(self._name, ex))
|
||||||
|
|
||||||
|
|
||||||
## replace caracter
|
## replace caracter
|
||||||
|
|
||||||
@@ -611,25 +828,7 @@ class WPimport:
|
|||||||
data = {"title":title, "content":bodyhtml, "status":"publish", "date": "{0}-{1}-{2}T{3}:00".format(time[2],month[time[1]],time[0], hour), "tags": listelement["tags"], "categories": listelement["categories"]}
|
data = {"title":title, "content":bodyhtml, "status":"publish", "date": "{0}-{1}-{2}T{3}:00".format(time[2],month[time[1]],time[0], hour), "tags": listelement["tags"], "categories": listelement["categories"]}
|
||||||
self._logger.debug("{0} : Data for post : |{1}| : {2}" .format(self._name, title, data))
|
self._logger.debug("{0} : Data for post : |{1}| : {2}" .format(self._name, title, data))
|
||||||
|
|
||||||
params = {"search":author, "per_page":100}
|
data["author"] = self._getAuthor(author)
|
||||||
try:
|
|
||||||
self._logger.info("{0} : Search author : {1}".format(self._name, author))
|
|
||||||
page = self._request.get("{1}://{0}/wp-json/wp/v2/users".format(self._wordpress, self._protocol), auth=self._basic, headers=self._headers_json, params=params)
|
|
||||||
self._logger.debug("{0} : End Search author : {1}".format(self._name, author))
|
|
||||||
self._logger.debug("{0} : Debug requests : {1}".format(self._name, page.content))
|
|
||||||
if page.status_code == 200:
|
|
||||||
self._logger.info("{0} : Get author id : {1}".format(self._name, result))
|
|
||||||
result = page.json()
|
|
||||||
for a in result:
|
|
||||||
data["author"] = a["id"]
|
|
||||||
else:
|
|
||||||
self._logger.error("{0} : Connection error with status code for get author : {1}".format(self._name, page.status_code))
|
|
||||||
self._logger.debug("{0} : {1}".format(page.content))
|
|
||||||
except ConnectionError as err:
|
|
||||||
self._logger.error("{0} : Connection error for get author : {1}".format(self._name, err))
|
|
||||||
exit(1)
|
|
||||||
except Exception as err:
|
|
||||||
self._logger.error("{0} : Exception error for get author : {1}".format(self._name, err))
|
|
||||||
page_is_exist = False
|
page_is_exist = False
|
||||||
|
|
||||||
for index in range(1,10):
|
for index in range(1,10):
|
||||||
|
394
lib/WPMenu.py
Normal file
394
lib/WPMenu.py
Normal file
@@ -0,0 +1,394 @@
|
|||||||
|
#!/usr/bin/python3
|
||||||
|
|
||||||
|
from bs4 import BeautifulSoup
|
||||||
|
from urllib.parse import urlparse
|
||||||
|
import requests, os, logging, re, json
|
||||||
|
from requests.adapters import HTTPAdapter
|
||||||
|
from requests.packages.urllib3.util.retry import Retry
|
||||||
|
|
||||||
|
class WPMenu:
|
||||||
|
# Constructor
|
||||||
|
def __init__(self, name="Thread-0", basic=None, canalblog="", wordpress="", logger=None, parser="html.parser", ssl_canalblog=True, ssl_wordpress=True):
|
||||||
|
self._name = name
|
||||||
|
self._basic = basic
|
||||||
|
self._canalblog = canalblog
|
||||||
|
self._wordpress = wordpress
|
||||||
|
self._logger = logger
|
||||||
|
self._parser = parser
|
||||||
|
self._headers_json = {'Content-Type': 'application/json; charset=utf-8', 'Accept':'application/json'}
|
||||||
|
self._protocol_wordpress = "https"
|
||||||
|
self._protocol_canalblog = "https"
|
||||||
|
self._directory = "backup"
|
||||||
|
|
||||||
|
if ssl_wordpress is False:
|
||||||
|
self._protocol_wordpress = "http"
|
||||||
|
if ssl_canalblog is False:
|
||||||
|
self._protocol_canalblog = "http"
|
||||||
|
self._request_canalblog = requests.Session()
|
||||||
|
self._request_wordpress = requests.Session()
|
||||||
|
|
||||||
|
retries = Retry(connect=10, read=10, redirect=5,
|
||||||
|
status_forcelist=[429, 500, 502, 503, 504], backoff_factor=2)
|
||||||
|
|
||||||
|
self._request_canalblog.mount('{0}://'.format(self._protocol_canalblog), HTTPAdapter(max_retries=retries))
|
||||||
|
self._request_wordpress.mount('{0}://'.format(self._protocol_wordpress), HTTPAdapter(max_retries=retries))
|
||||||
|
|
||||||
|
|
||||||
|
# Destructor
|
||||||
|
def __del__(self):
|
||||||
|
print("{0} : Import finished for {1}".format(self._name, self._wordpress))
|
||||||
|
|
||||||
|
# Public method
|
||||||
|
|
||||||
|
## From file
|
||||||
|
|
||||||
|
def fromFile(self, files):
|
||||||
|
if os.path.exists(files):
|
||||||
|
with open(files, 'r') as f:
|
||||||
|
self._logger.info("{0} : File is being processed : {1}".format(self._name, files))
|
||||||
|
content = f.read()
|
||||||
|
self._menu(content)
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : File isn't exist : {1}".format(self._name, files))
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
## Get from URL
|
||||||
|
|
||||||
|
def fromUrl(self, canalblog):
|
||||||
|
self._canalblog = canalblog
|
||||||
|
try:
|
||||||
|
o = urlparse(canalblog)
|
||||||
|
o = o._replace(scheme=self._protocol_canalblog)
|
||||||
|
i = o.geturl().replace(":///", "://")
|
||||||
|
page = self._request_canalblog.get(i)
|
||||||
|
if page.status_code == 200:
|
||||||
|
self._logger.info("{0} : Page web is being processed : {1}".format(self._name, i))
|
||||||
|
self._menu(page.content)
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : index didn't get due status code : {1}".format(self._name, page.status_code))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, page.content))
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for get url {1} : {2}".format(self._name, canalblog, err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for get url {1} : {2}".format(self._name, canalblog, err))
|
||||||
|
|
||||||
|
## replace caracter
|
||||||
|
|
||||||
|
def _replaceCaracter(self, title_rendered):
|
||||||
|
list_replace = {'’': "'", '–': '-', '…': '...', '« ': '"', ' »': '"', '« ': '"', ' »': '"', '’': "'", '"‘': "'"}
|
||||||
|
for old, new in list_replace.items():
|
||||||
|
title_rendered = title_rendered.replace(old, new)
|
||||||
|
return title_rendered
|
||||||
|
|
||||||
|
def _getIdfromTitlePost(self, content):
|
||||||
|
idMenu = {"id":0, "type":"", "link":""}
|
||||||
|
soup = BeautifulSoup(content, self._parser)
|
||||||
|
articletitle = soup.find_all("h2", class_="articletitle")
|
||||||
|
if len(articletitle) > 0:
|
||||||
|
articletitle = articletitle[0].get_text()
|
||||||
|
search = "posts"
|
||||||
|
post_type = "post"
|
||||||
|
if len(articletitle) == 0:
|
||||||
|
articletitle = soup.find_all("div", class_="albumbody")
|
||||||
|
if len(articletitle) > 0:
|
||||||
|
articletitle = articletitle[0].find("h2").get_text()
|
||||||
|
search = "pages"
|
||||||
|
post_type = "page"
|
||||||
|
|
||||||
|
exist = False
|
||||||
|
for index in range(1,10):
|
||||||
|
if exist is False:
|
||||||
|
params = {"search":articletitle, "per_page":100, "page":index}
|
||||||
|
try:
|
||||||
|
self._logger.debug("{0} : Get Url for {3} : {1} {2}".format(self._name, "{1}://{0}/wp-json/wp/v2/{2}".format(self._wordpress, self._protocol_wordpress, search), params, search))
|
||||||
|
page = self._request_wordpress.get("{1}://{0}/wp-json/wp/v2/{2}".format(self._wordpress, self._protocol_wordpress, search), auth=self._basic, params=params)
|
||||||
|
if page.status_code == 200:
|
||||||
|
result = page.json()
|
||||||
|
self._logger.info("{0} : Get content {2} : {1}".format(self._name, len(result), search))
|
||||||
|
if len(result) > 0:
|
||||||
|
for i in result:
|
||||||
|
title_rendered = i["title"]["rendered"]
|
||||||
|
if len(articletitle) != len(title_rendered):
|
||||||
|
title_rendered = self._replaceCaracter(title_rendered)
|
||||||
|
self._logger.debug("{0} : comparaison debug {1} {2}".format(self._name, articletitle, title_rendered))
|
||||||
|
if articletitle == title_rendered:
|
||||||
|
self._logger.debug("{0} : get {2} id : {1}".format(self._name, i, search))
|
||||||
|
idMenu = {"id":i["id"], "type":post_type, "link": i["link"]}
|
||||||
|
exist = True
|
||||||
|
else:
|
||||||
|
self._logger.debug("{0} : {2} {1}".format(self._name, result, len(result)))
|
||||||
|
break
|
||||||
|
elif page.status_code == 400:
|
||||||
|
self._logger.debug("{0} : {2} {1}".format(self._name, page.content, page.status_code))
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : Post didn't get due status code : {1}".format(self._name, page.status_code))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, page.content))
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for get content : {1}".format(self._name, err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for get content : {1} ".format(self._name, err))
|
||||||
|
return idMenu
|
||||||
|
|
||||||
|
def _getIdFromPost(self, href):
|
||||||
|
idMenu = {"id":0, "type":"", "link":""}
|
||||||
|
o = urlparse(href)
|
||||||
|
if len(o.netloc) > 0:
|
||||||
|
try:
|
||||||
|
page = self._request_canalblog.get(href)
|
||||||
|
if page.status_code == 200:
|
||||||
|
self._logger.info("{0} : Get content : {1}".format(self._name, href))
|
||||||
|
idMenu = self._getIdfromTitlePost(page.content)
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : {2} didn't get due status code : {1}".format(self._name, page.status_code, href))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, page.content))
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for get url {1} : {2}".format(self._name, href, err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for get url {1} : {2}".format(self._name, href, err))
|
||||||
|
else:
|
||||||
|
if os.path.exists("{0}/..{1}".format(self._directory, o.path)):
|
||||||
|
try:
|
||||||
|
content = open("{0}/..{1}".format(self._directory, o.path), "r")
|
||||||
|
idMenu = self._getIdfromTitlePost(content)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for get file content {1} : {2}".format(self._name, href, err))
|
||||||
|
return idMenu
|
||||||
|
|
||||||
|
|
||||||
|
def _getIdFromReverse(self, title, href):
|
||||||
|
self._logger.info("{0} : get title {1} from href {2}".format(self._name, title, href))
|
||||||
|
idMenu = {"id":0, "type":"", "link":""}
|
||||||
|
if href != "#":
|
||||||
|
title = href[::-1]
|
||||||
|
second_title = title.split("/")[2]
|
||||||
|
second_title = second_title[::-1]
|
||||||
|
link = title.split("/")[0]
|
||||||
|
link = link[::-1]
|
||||||
|
title = title.split("/")[1]
|
||||||
|
title = title[::-1]
|
||||||
|
self._logger.info("{0} link {1} title {2}".format(self._name, link, title))
|
||||||
|
if link == "index.html":
|
||||||
|
if second_title == "albums":
|
||||||
|
idMenu = self._getIdFromPost(href)
|
||||||
|
else:
|
||||||
|
idMenu = self._getId(title)
|
||||||
|
else:
|
||||||
|
idMenu = self._getIdFromPost(href)
|
||||||
|
|
||||||
|
return idMenu
|
||||||
|
|
||||||
|
def _getId(self, title):
|
||||||
|
idMenu = {"id": 0, "type":"", "link":""}
|
||||||
|
exist = False
|
||||||
|
if exist is False:
|
||||||
|
for i in ["categories", "tags"]:
|
||||||
|
typeId = "category"
|
||||||
|
if i == "tags":
|
||||||
|
typeId = "tag"
|
||||||
|
for index in range(1,10):
|
||||||
|
try:
|
||||||
|
params = {"search":title, "per_page":"100", "page":index}
|
||||||
|
self._logger.info("{0} Get menu {1} {2} {3}".format(self._name, "{2}://{0}/wp-json/wp/v2/{1}".format(self._wordpress, i, self._protocol_wordpress), index, title))
|
||||||
|
page = self._request_wordpress.get("{2}://{0}/wp-json/wp/v2/{1}".format(self._wordpress, i, self._protocol_wordpress), auth=self._basic, params=params)
|
||||||
|
if page.status_code == 200:
|
||||||
|
result = page.json()
|
||||||
|
if len(result) > 0:
|
||||||
|
for j in result:
|
||||||
|
self._logger.info("{0} info : {1} {2} {3}".format(self._name, j["name"], j["slug"], title))
|
||||||
|
if j["name"] == title or j["slug"] == title:
|
||||||
|
self._logger.info("{0} : comparaison ok : {1} {2}".format(self._name, j["id"], i))
|
||||||
|
|
||||||
|
idMenu = {"id": j["id"], "type": typeId, "link": j["link"]}
|
||||||
|
exist = True
|
||||||
|
else:
|
||||||
|
break
|
||||||
|
elif page.status_code == 400:
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : {2} didn't get due status code : {1}".format(self._name, page.status_code, i))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, page.content))
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for get url {1} : {2}".format(self._name, "{2}://{0}/wp-json/wp/v2/{1}".format(self._wordpress, i, self._protocol_wordpress), err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for get url {1} : {2}".format(self._name, "{2}://{0}/wp-json/wp/v2/{1}".format(self._wordpress, i, self._protocol_wordpress), err))
|
||||||
|
return idMenu
|
||||||
|
|
||||||
|
def _menu(self, content):
|
||||||
|
|
||||||
|
soup = BeautifulSoup(content, self._parser)
|
||||||
|
ul = soup.find("ul", id="listsmooth")
|
||||||
|
menu = list()
|
||||||
|
children = list()
|
||||||
|
for anchor in ul.find_all("li"):
|
||||||
|
|
||||||
|
parent = anchor.find("a").get_text().replace(" \xa0", "")
|
||||||
|
href = anchor.find("a").get("href")
|
||||||
|
if href == "{0}://{1}/".format(self._protocol_canalblog, self._canalblog):
|
||||||
|
parent = "home"
|
||||||
|
itemMenu = {"id":"", "type":"", "title": parent, "link":"", "href":href, "children":list()}
|
||||||
|
if href == "#":
|
||||||
|
li = anchor.find("ul").find_all("li")
|
||||||
|
for child in li:
|
||||||
|
a = child.find("a")
|
||||||
|
href = a.get("href")
|
||||||
|
self._logger.info("{0} Parent {1} : Child {2}".format(self._name, parent, a.get_text()))
|
||||||
|
children.append({"title": a.get_text(), "parent": parent, "href":href, "link":""})
|
||||||
|
menu.append(itemMenu)
|
||||||
|
|
||||||
|
|
||||||
|
for i in range(0, len(children)):
|
||||||
|
self._logger.info("{0} : Child {1} {2}".format(self._name, children[i], i))
|
||||||
|
for j in range(0, len(menu)):
|
||||||
|
if j < len(menu):
|
||||||
|
if menu[j]["title"] == children[i]["title"]:
|
||||||
|
self._logger.info("{0} : Parent {1} {2}".format(self._name, menu[j], j))
|
||||||
|
del menu[j]
|
||||||
|
|
||||||
|
for j in range(0, len(menu)):
|
||||||
|
self._logger.info("{0} : Children for : {1}".format(self._name, menu[j]["title"]))
|
||||||
|
if menu[j]["title"] == children[i]["parent"]:
|
||||||
|
menu[j]["children"].append({"id":"", "type":"", "title":children[i]["title"], "parent": children[i]["parent"], "link":"", "href":children[i]["href"]})
|
||||||
|
|
||||||
|
for i in range(0, len(menu)):
|
||||||
|
self._logger.info("{0} : Menu {1} {2}".format(self._name, menu[i]["title"], len(menu[i]["children"])))
|
||||||
|
if menu[i]["title"] != "home":
|
||||||
|
for j in range(0, len(menu[i]["children"])):
|
||||||
|
idMenu = self._getId(menu[i]["children"][j]["title"])
|
||||||
|
if idMenu["id"] == 0:
|
||||||
|
self._logger.debug("{0} : content children {1}".format(self._name, menu[i]["children"][j]))
|
||||||
|
idMenu = self._getIdFromReverse(menu[i]["children"][j]["title"], menu[i]["children"][j]["href"])
|
||||||
|
if idMenu["id"] != 0:
|
||||||
|
menu[i]["children"][j] = {"id":idMenu["id"], "type": idMenu["type"], "link": idMenu["link"], "title": menu[i]["children"][j]["title"], "parent": menu[i]["children"][j]["parent"]}
|
||||||
|
idMenu = self._getId(menu[i]["title"])
|
||||||
|
self._logger.debug("{0} : content parent {1}".format(self._name, menu[i]))
|
||||||
|
self._logger.debug("{0} : content idMenu {1}".format(self._name, idMenu))
|
||||||
|
|
||||||
|
if idMenu["id"] == 0:
|
||||||
|
idMenu = self._getIdFromReverse(menu[i]["title"], menu[i]["href"])
|
||||||
|
if idMenu["id"] != 0:
|
||||||
|
menu[i] = {"id":idMenu["id"], "type": idMenu["type"], "title":menu[i]["title"], "link":idMenu["link"], "children": menu[i]["children"]}
|
||||||
|
|
||||||
|
self._createMenu(menu)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def _createItemMenu(self, idMenu, itemMenu, parent):
|
||||||
|
idItemMenu = 0
|
||||||
|
self._logger.info("{0} : Create item menu from API Wordpress : {1}".format(self._name, self._wordpress))
|
||||||
|
try:
|
||||||
|
params = {"search": itemMenu["title"], "menus": idMenu}
|
||||||
|
page = self._request_wordpress.get("{1}://{0}/wp-json/wp/v2/menu-items".format(self._wordpress, self._protocol_wordpress), auth=self._basic, params=params)
|
||||||
|
if page.status_code == 200:
|
||||||
|
result = page.json()
|
||||||
|
for i in result:
|
||||||
|
if self._replaceCaracter(i["title"]["rendered"]) == itemMenu["title"]:
|
||||||
|
idItemMenu = int(i["id"])
|
||||||
|
self._logger.info("{0} : Length of result for item menus : {1}".format(self._name, len(result)))
|
||||||
|
url = "{1}://{0}/wp-json/wp/v2/menu-items".format(self._wordpress, self._protocol_wordpress)
|
||||||
|
if idItemMenu != 0:
|
||||||
|
url = "{1}://{0}/wp-json/wp/v2/menu-items/{2}".format(self._wordpress, self._protocol_wordpress, idItemMenu)
|
||||||
|
|
||||||
|
try:
|
||||||
|
objectt = itemMenu["type"]
|
||||||
|
if objectt == "tag":
|
||||||
|
objectt = "post_tag"
|
||||||
|
data = {"title": itemMenu["title"], "status": "publish", "parent":parent, "menus":idMenu, "url":"#"}
|
||||||
|
if itemMenu["title"] == "home":
|
||||||
|
data = {"title": itemMenu["title"], "status": "publish", "parent":parent, "menus":idMenu, "url":"{0}://{1}".format(self._protocol_wordpress, self._wordpress)}
|
||||||
|
if type(itemMenu["id"]) is str:
|
||||||
|
if len(itemMenu["id"]) > 0:
|
||||||
|
data = {"title": itemMenu["title"], "status": "publish", "parent":parent, "url": itemMenu["link"], "menus":idMenu, "object":objectt, "object_id":int(itemMenu["id"])}
|
||||||
|
elif type(itemMenu["id"]) is int:
|
||||||
|
data = {"title": itemMenu["title"], "status": "publish", "parent":parent, "url": itemMenu["link"], "menus":idMenu, "object":objectt, "object_id":itemMenu["id"]}
|
||||||
|
self._logger.debug("{0} : data for create/update : {1}".format(self._name, data))
|
||||||
|
page = self._request_wordpress.post(url, auth=self._basic, headers=self._headers_json, data=json.dumps(data))
|
||||||
|
|
||||||
|
if page.status_code in [201, 200]:
|
||||||
|
result = page.json()
|
||||||
|
idItemMenu = int(result["id"])
|
||||||
|
self._logger.info("{0} : create/update item menu : {1}".format(self._name, itemMenu["title"]))
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : Create menu items for {2} didn't get due status code : {1}".format(self._name, page.status_code, itemMenu["title"]))
|
||||||
|
self._logger.debug("{0} : {1} {2}".format(self._name, page.content, itemMenu))
|
||||||
|
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for create item menu {1} : {2}".format(self._name, "{1}://{0}/wp-json/wp/v2/menu-items".format(self._wordpress, self._protocol_wordpress), err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for create item menu {1} : {2}".format(self._name, "{1}://{0}/wp-json/wp/v2/menu-items".format(self._wordpress, self._protocol_wordpress), err))
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : Get menu items didn't get due status code : {1}".format(self._name, page.status_code))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, page.content))
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for get item menus {1} : {2}".format(self._name, "{1}://{0}/wp-json/wp/v2/menu-items".format(self._wordpress, self._protocol_wordpress), err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for get item menus {1} : {2}".format(self._name, "{1}://{0}/wp-json/wp/v2/menu-items".format(self._wordpress, self._protocol_wordpress), err))
|
||||||
|
return idItemMenu
|
||||||
|
|
||||||
|
def _createMenu(self, menu):
|
||||||
|
title = "Menu {0}".format(self._wordpress)
|
||||||
|
self._logger.info("{0} : Create menu from API Wordpress : {1}".format(self._name, title))
|
||||||
|
try:
|
||||||
|
params = {"search": title}
|
||||||
|
page = self._request_wordpress.get("{1}://{0}/wp-json/wp/v2/menus".format(self._wordpress, self._protocol_wordpress), auth=self._basic, params=params)
|
||||||
|
if page.status_code == 200:
|
||||||
|
result = page.json()
|
||||||
|
self._logger.info("{0} : Get content menus : {1}".format(self._name, len(result)))
|
||||||
|
idMenu = 0
|
||||||
|
if len(result) == 0:
|
||||||
|
|
||||||
|
self._logger.info("{0} : Create menu : {1}".format(self._name, title))
|
||||||
|
data = {"name": title}
|
||||||
|
try:
|
||||||
|
page = self._request_wordpress.post("{1}://{0}/wp-json/wp/v2/menus".format(self._wordpress, self._protocol_wordpress), auth=self._basic, headers=self._headers_json, data=json.dumps(data))
|
||||||
|
if page.status_code == 201:
|
||||||
|
result = page.json()
|
||||||
|
self._logger.debug("{0} : Get menus : {1}".format(self._name, result))
|
||||||
|
|
||||||
|
if len(result) > 0:
|
||||||
|
idMenu = result["id"]
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : Post menu didn't get due status code : {1}".format(self._name, page.status_code))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, page.content))
|
||||||
|
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for create menu {1} : {2}".format(self._name, "{1}://{0}/wp-json/wp/v2/menus".format(self._wordpress, self._protocol_wordpress), err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for get menu {1} : {2}".format(self._name, "{1}://{0}/wp-json/wp/v2/menus".format(self._wordpress, self._protocol_wordpress), err))
|
||||||
|
else:
|
||||||
|
self._logger.debug("{0} : Get menus : {1}".format(self._name, result))
|
||||||
|
for i in result:
|
||||||
|
self._logger.debug("{0} : List menus : {1}".format(self._name, i))
|
||||||
|
if i["name"] == title:
|
||||||
|
idMenu = i["id"]
|
||||||
|
self._logger.info("{0} : Get ID menus : {1}".format(self._name, idMenu))
|
||||||
|
self._addItemMenu(menu, idMenu)
|
||||||
|
else:
|
||||||
|
self._logger.error("{0} : Get menu didn't get due status code : {1}".format(self._name, page.status_code))
|
||||||
|
self._logger.debug("{0} : {1}".format(self._name, page.content))
|
||||||
|
|
||||||
|
except ConnectionError as err:
|
||||||
|
self._logger.error("{0} : Connection error for get menu {1} : {2}".format(self._name, "{1}://{0}/wp-json/wp/v2/menus".format(self._wordpress, self._protocol_wordpress), err))
|
||||||
|
exit(1)
|
||||||
|
except Exception as err:
|
||||||
|
self._logger.error("{0} : Exception error for get menu {1} : {2}".format(self._name, "{1}://{0}/wp-json/wp/v2/menus".format(self._wordpress, self._protocol_wordpress), err))
|
||||||
|
|
||||||
|
|
||||||
|
def _addItemMenu(self, menu, idMenu):
|
||||||
|
self._logger.info("{0} : add item to menu : {1}".format(self._name, idMenu))
|
||||||
|
parent = 0
|
||||||
|
for i in menu:
|
||||||
|
parent = 0
|
||||||
|
self._logger.debug("{0} : debug create item menu : {1}".format(self._name, i))
|
||||||
|
parent = self._createItemMenu(idMenu, i, parent)
|
||||||
|
for j in i["children"]:
|
||||||
|
self._createItemMenu(idMenu, j, parent)
|
Reference in New Issue
Block a user