v4l3n71n/web_scrap
1
0
Fork 0
Code Issues Pull Requests Projects Releases 10 Wiki Activity

menu wordpress 75%

Browse Source
This commit is contained in:
Valentin CZERYBA 2023-07-27 00:39:15 +02:00
parent b7493206a2
commit 5399b12133
1 changed files with 20 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
lib/WPMenu.py
View File

@ -273,7 +273,7 @@ class WPMenu:
result = page.json() result = page.json()
for i in result: for i in result:
if i["title"]["rendered"] == itemMenu["title"]: if i["title"]["rendered"] == itemMenu["title"]:
idItemMenu = i["id"] idItemMenu = int(i["id"])
self._logger.info("{0} : Length of result for item menus : {1}".format(self._name, len(result))) 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) url = "{1}://{0}/wp-json/wp/v2/menu-items".format(self._wordpress, self._protocol_wordpress)
if idItemMenu != 0: if idItemMenu != 0:
@ -283,28 +283,28 @@ class WPMenu:
objectt = itemMenu["type"] objectt = itemMenu["type"]
if objectt == "tag": if objectt == "tag":
objectt = "post_tag" objectt = "post_tag"
data = {"title": itemMenu["title"], "status": "publish", "parent":parent, "menus":idMenu, "object":objectt, "object_id":itemMenu["id"]} 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)) page = self._request_wordpress.post(url, auth=self._basic, headers=self._headers_json, data=json.dumps(data))
if page.status_code == 201: if page.status_code in [201, 200]:
result = page.json() result = page.json()
idItemMenu = result["id"] idItemMenu = int(result["id"])
self._logger.info("{0} : create item menu : {1}".format(self._name, itemMenu["title"])) self._logger.info("{0} : create/update item menu : {1}".format(self._name, itemMenu["title"]))
elif page.status_code == 200:
result = page.json()
for i in result:
if result["title"]["rendered"] == itemMenu["title"]:
idItemMenu = i["id"]
self._logger.info("{0} : update item menu : {1}".format(self._name, itemMenu["title"]))
else: else:
self._logger.error("{0} : Create menu items didn't get due status code : {1}".format(self._name, page.status_code)) 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}".format(self._name, page.content)) self._logger.debug("{0} : {1} {2}".format(self._name, page.content, itemMenu))
except ConnectionError as err: 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)) 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) exit(1)
except Exception as err: except Exception as err:
self._logger.error("{0} : Exception error for get item menu {1} : {2}".format(self._name, "{1}://{0}/wp-json/wp/v2/menu-items".format(self._wordpress, self._protocol_wordpress), 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: else:
self._logger.error("{0} : Get menu items didn't get due status code : {1}".format(self._name, page.status_code)) 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)) self._logger.debug("{0} : {1}".format(self._name, page.content))
@ -353,6 +353,7 @@ class WPMenu:
if i["name"] == title: if i["name"] == title:
idMenu = i["id"] idMenu = i["id"]
self._logger.info("{0} : Get ID menus : {1}".format(self._name, idMenu)) self._logger.info("{0} : Get ID menus : {1}".format(self._name, idMenu))
self._addItemMenu(menu, idMenu)
else: else:
self._logger.error("{0} : Get menu didn't get due status code : {1}".format(self._name, page.status_code)) 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)) self._logger.debug("{0} : {1}".format(self._name, page.content))
@ -364,10 +365,11 @@ class WPMenu:
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)) 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): def _addItemMenu(self, menu, idMenu):
self._logger.info("{0} : add item to menu : {1}".format(self._name, idMenu)) self._logger.info("{0} : add item to menu : {1}".format(self._name, idMenu))
parent = 0
for i in menu: for i in menu:
self._logger.debug("{0} : debug create item menu : {1}".format(self._name, i))
parent = self._createItemMenu(idMenu, i, parent) parent = self._createItemMenu(idMenu, i, parent)
for j in menu["children"]: for j in i["children"]:
self._createItemMenu(idMenu, i, parent) self._createItemMenu(idMenu, j, parent)