add variable

lib/WPRemove.py

@@ -8,14 +8,15 @@ from requests.packages.urllib3.util.retry import Retry
 
 class WPRemove:
     # Constructor
-    def __init__(self, name="Thread-0", basic=None, wordpress="", logger=None, ssl_wordpress=True):
+    def __init__(self, index_name=1, number_thread=1, basic=None, wordpress="", logger=None, ssl_wordpress=True):
-        self._name = name
         self._basic = basic
         self._wordpress = wordpress
         self._logger = logger
         self._headers_json = {'Content-Type': 'application/json', 'Accept':'application/json'}
-        
+        self._name =  "Thread-{0}".format(index_name)
+        self._index_thread = index_name
         self._protocol = "https"
+        self._number_thread = number_thread
         if ssl_wordpress is False:
             self._protocol = "http"
         self._request = requests.Session()
@@ -32,6 +33,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
     
@@ -55,8 +74,7 @@ class WPRemove:
             self._logger.info("{0} : List {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), auth=self._basic, params=params, headers=self._headers_json)
-        except Exception as err:
+        
-            self._logger.error("{0} : Connection error for list {1} to remove : {2}".format(self._name, composant, err))
             if r.status_code == 200:
                 result = r.json()
                 if len(result) > 0:
@@ -76,3 +94,5 @@ class WPRemove:
             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 Exception as err:
+            self._logger.error("{0} : Connection error for list {1} to remove : {2}".format(self._name, composant, err))