From 6401692d0d62bc36632c61161de4bd61be606551 Mon Sep 17 00:00:00 2001 From: Valentin CZERYBA Date: Fri, 25 Aug 2023 23:46:43 +0200 Subject: [PATCH] finish menu with album import --- import_export_canalblog.py | 3 +-- lib/WPMenu.py | 10 ++++++---- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/import_export_canalblog.py b/import_export_canalblog.py index 1059440..78fdb0f 100644 --- a/import_export_canalblog.py +++ b/import_export_canalblog.py @@ -170,7 +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-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("--author", help="Define author for page album", default="") + import_parser.add_argument("--author-album", dest=author, help="Define author for page album", default="") @@ -264,7 +264,6 @@ if __name__ == '__main__': for i in wordpress: importWp.setUrl(i) importWp.fromFile(files=args.file.split(",")) - exit(0) 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: diff --git a/lib/WPMenu.py b/lib/WPMenu.py index 4d868d1..bc7c8ea 100644 --- a/lib/WPMenu.py +++ b/lib/WPMenu.py @@ -91,9 +91,11 @@ class WPMenu: search = "posts" post_type = "post" if len(articletitle) == 0: - articletitle = soup.find("div", class_="albumbody").find("h2").get_text() - search = "pages" - post_type = "page" + 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): @@ -171,7 +173,7 @@ class WPMenu: title = title[::-1] self._logger.info("{0} link {1} title {2}".format(self._name, link, title)) if link == "index.html": - if second_title == "archives" or second_title == "albums": + if second_title == "albums": idMenu = self._getIdFromPost(href) else: idMenu = self._getId(title)