add class + wordpress

insert_wordpress.py

@@ -5,27 +5,33 @@ from requests.auth import HTTPBasicAuth
 from getpass import getpass
 import requests, os, argparse, logging, re
 
-def insertWordpress(file, basic):
+class WPimport:
+    
+    def __init__(self, basic, wordpress):
+        self.basic = basic
+        self.wordpress = wordpress
+
+    def fromFile(self, file):
+        with open(file, 'r') as f:
+            contents = f.read()
+            self.insertWordpress(contents)
+
+    def insertWordpress(self, content):
         tags = []
         month = {"janvier":"01", "février": "02", "mars": "03", "avril":"04", "mai": "05", "juin": "06", "juillet": "07", "août": "08", "septembre": "09", "octobre": "10", "novembre": "11", "décembre": "12"}
-
         liste = ["categories", "tags"]
         elements = {}
         element = {}
         listelement = {}
 
         for i in liste:
-        page = requests.get("http://localhost:8080/wp-json/wp/v2/{0}".format(i))
+            page = requests.get("http://{0}/wp-json/wp/v2/{1}".format(self.wordpress,i))
             if page.status_code == 200:
                 elements[i] = page.json()
                 element[i] = []
                 listelement[i] = []
 
-    
+        soup = BeautifulSoup(content, 'html.parser')
-    with open(file, 'r') as f:
-        contents = f.read()
-
-        soup = BeautifulSoup(contents, 'html.parser')
         articletitle = soup.find_all("h2", class_="articletitle")
         articlebody = soup.find_all("div", class_="articlebody")
         articledate = soup.find_all("span", class_="articledate")
@@ -63,14 +69,11 @@ def insertWordpress(file, basic):
                         array = listelement[i].append(k["id"])
                 if element_exist is False:
                     data = {"name": j}
-                    page = requests.post("http://localhost:8080/wp-json/wp/v2/{0}".format(i), auth=basic, data=data)
+                    page = requests.post("http://{0}/wp-json/wp/v2/{1}".format(self.wordpress, i), auth=self.basic, data=data)
                     if page.status_code == 201:
                         result = page.json()
                         listelement[i].append(result["id"])
 
-
-      
-
         title = articletitle[0].text
         author = articleacreator[0].text.lower()        
         body = articlebody[0].find_all("p")
@@ -86,14 +89,13 @@ def insertWordpress(file, basic):
 
         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"]}
         params = {"search":author}
-        page = requests.get("http://localhost:8080/wp-json/wp/v2/users", auth=basic, params=params)
+        page = requests.get("http://{0}/wp-json/wp/v2/users".format(self.wordpress), auth=self.basic, params=params)
         if page.status_code == 200:
             result = page.json()
             data["author"] = result[0]["id"]
 
-    
         params = {"search":title}
-    page = requests.get("http://localhost:8080/wp-json/wp/v2/posts", auth=basic, params=params)
+        page = requests.get("http://{0}/wp-json/wp/v2/posts".format(self.wordpress), auth=self.basic, params=params)
         page_exist = True
         if page.status_code:
             result = page.json()
@@ -103,24 +105,28 @@ def insertWordpress(file, basic):
                 print("La page {0} existe deja".format(title))
 
         if page_exist == False:
-        page = requests.post("http://localhost:8080/wp-json/wp/v2/posts", auth=basic, data=data)
+            page = requests.post("http://{0}/wp-json/wp/v2/posts".format(self.wordpress), auth=self.basic, data=data)
             if page.status_code == 201:
                 result = page.json()
                 print("Article ajoute : {0}".format(result["title"]["raw"]))
                 print(comment_post)
                 for i in comment_post:
                     data = {"post": result["id"], "content": i["content"], "date": i["date"], "author_name": i["author"]}
-                page = requests.post("http://localhost:8080/wp-json/wp/v2/comments", auth=basic, data=data)
+                    page = requests.post("http://{0}/wp-json/wp/v2/comments".format(self.wordpress), auth=self.basic, data=data)
                     print(page.status_code)
                     if page.status_code == 201:
                         print("Commentaire ajoute pour {0}".format(result["title"]["raw"]))
 
 
 
+    
+
+
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument("--user", help="wordpress user", required=True)
     parser.add_argument("--file", help="HTML file", required=True)
+    parser.add_argument("--wordpress", help="URL Wordpress", required=True)
     args = parser.parse_args()
     password = getpass()
     if len(password) == 0:
@@ -128,4 +134,5 @@ if __name__ == '__main__':
         exit(1)
 
     basic = HTTPBasicAuth(args.user, password)
-    insertWordpress(args.file, basic)
+    importWp = WPimport(basic)
+    importWp.fromFile(args.file)