diff --git a/lib/WPRemove.py b/lib/WPRemove.py index 8d38ca7..1e5799e 100644 --- a/lib/WPRemove.py +++ b/lib/WPRemove.py @@ -32,6 +32,24 @@ class WPRemove: # Public method + def _getCount(self, composant): + count = 0 + try: + params = {"per_page":1} + self._logger.info("{0} : Get count {2} to remove for url : {1}".format(self._name, self._wordpress, composant)) + r = self._request.get("{2}://{0}/wp-json/wp/v2/{1}".format(self._wordpress, composant, self._protocol), params=params, auth=self._basic, headers=self._headers_json) + if r.status_code == 200: + count = int(r.headers["X-WP-Total"]) + else: + self._logger.error("{0} : Error for list to remove {1} due status code {2}".format(self._name, composant, r.status_code)) + self._logger.debug("{0} : Content error for {1} : {2}".format(self._name, composant, r.content)) + except ConnectionError as err: + self._logger.error("{0} : Connection error for list {1} to remove : {2}".format(self._name, composant, err)) + exit(1) + except Exception as err: + self._logger.error("{0} : Exception error for list {1} to remove : {2}".format(self._name, composant, err)) + return count + def setUrl(self, wordpress): self._wordpress = wordpress