76 lines
3.2 KiB
Python
76 lines
3.2 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
|
||
|
|
from bs4 import BeautifulSoup
|
||
|
|
from urllib.parse import urlparse
|
||
|
|
import requests, os, logging, re, json
|
||
|
|
from requests.adapters import HTTPAdapter
|
||
|
|
from requests.packages.urllib3.util.retry import Retry
|
||
|
|
|
||
|
|
class WPRemove:
|
||
|
|
# Constructor
|
||
|
|
def __init__(self, name="Thread-0", basic=None, wordpress="", logger=None):
|
||
|
|
self._name = name
|
||
|
|
self._basic = basic
|
||
|
|
self._wordpress = wordpress
|
||
|
|
self._logger = logger
|
||
|
|
self._headers_json = {'Content-Type': 'application/json', 'Accept':'application/json'}
|
||
|
|
|
||
|
|
self._request = requests.Session()
|
||
|
|
|
||
|
|
retries = Retry(connect=10, read=10, redirect=5,
|
||
|
|
status_forcelist=[429, 500, 502, 503, 504], backoff_factor=2)
|
||
|
|
|
||
|
|
self._request.mount('http://', HTTPAdapter(max_retries=retries))
|
||
|
|
|
||
|
|
|
||
|
|
# Destructor
|
||
|
|
def __del__(self):
|
||
|
|
print("{0} : Import finished for {1}".format(self._name, self._wordpress))
|
||
|
|
|
||
|
|
# Public method
|
||
|
|
|
||
|
|
def setUrl(self, wordpress):
|
||
|
|
self._wordpress = wordpress
|
||
|
|
|
||
|
|
def cleanPosts(self):
|
||
|
|
self._removeAll("posts")
|
||
|
|
|
||
|
|
def cleanTags(self):
|
||
|
|
self._removeAll("tags")
|
||
|
|
|
||
|
|
def cleanCategories(self):
|
||
|
|
self._removeAll("categories")
|
||
|
|
|
||
|
|
def cleanMedia(self):
|
||
|
|
self._removeAll("media")
|
||
|
|
|
||
|
|
# Private method
|
||
|
|
|
||
|
|
def _removeAll(self, composant):
|
||
|
|
params = {"per_page":100}
|
||
|
|
try:
|
||
|
|
self._logger.info("{0} : List {2} to remove for url : {1}".format(self._name, self._wordpress, composant))
|
||
|
|
|
||
|
|
r = self._request.get("http://{0}/wp-json/wp/v2/{1}".format(self._wordpress, composant), 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:
|
||
|
|
for i in result:
|
||
|
|
self._logger.info("{0} : Remove {2} for url {1} : {3}".format(self._name, self._wordpress, composant, i["title"]["rendered"]))
|
||
|
|
params = {"force":1}
|
||
|
|
try:
|
||
|
|
r = self._request.delete("http://{0}/wp-json/wp/v2/{1}/{2}".format(self._wordpress, composant, i["id"]), auth=self._basic, headers=self._headers_json , params=params)
|
||
|
|
if r.status_code == 200:
|
||
|
|
self._logger.info("{0} : Post removed for URL {1} {2} : {3}".format(self._name, self._wordpress, composant, i["title"]["rendered"]))
|
||
|
|
else:
|
||
|
|
self._logger.error("{0} : Connection error for post {1} {2} {3} with status code {4}".format(self._name, self._wordpress, composant, i["title"]["rendered"], r.status_code))
|
||
|
|
except Exception as err:
|
||
|
|
self._logger.error("{0} : Connection error for {1} remove : {2}".format(self._name, composant, err))
|
||
|
|
exit(1)
|
||
|
|
self._removeAll(composant)
|
||
|
|
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))
|