add '--ignore-certificate' option

Author: SkypLabs

README.md

@@ -4,7 +4,7 @@ This Python script is made for doing a complete backup of your Wordpress blog's
 
 ## Dependancies
 
- * Python 2.7
+ * Python >= 2.7.9
  * [Mechanize][1] package
 
 ### On Fedora
@@ -21,8 +21,9 @@ This Python script is made for doing a complete backup of your Wordpress blog's
 
 ## How to
 
-	usage: wp-backup-data.py [-h] [-u USER] [-p PASSWORD] [-P] [-a ADDRESS]
-							 [-d DIRECTORY] [--http] [--https] [-v]
+	usage: wp_backup_data.py [-h] [-u USER] [-p PASSWORD] [-P] [-a ADDRESS]
+							 [-d DIRECTORY] [--http] [--https]
+							 [--ignore-certificate] [-v]
 
 	Do a backup of your WordPress data
 
@@ -40,6 +41,7 @@ This Python script is made for doing a complete backup of your Wordpress blog's
 							directory where the backup file will be stored
 	  --http                use HTTP as protocol
 	  --https               use HTTPS as protocol (default)
+	  --ignore-certificate  ignore invalid certificates
 	  -v, --version         show program's version number and exit
 
 	Example: ./wp-backup-data.py -a blog.example.net -u user -P

wp_backup_data.py

@@ -8,7 +8,13 @@
 from getpass import getpass
 from os import access, W_OK
 from os.path import isdir
-from sys import exit, stderr, stdout
+from sys import exit, stderr, stdout, version_info
+
+if version_info[2] > 8:
+	from ssl import CertificateError
+else:
+	class CertificateError(Exception):
+		pass
 
 class check_dir(Action):
 	def __call__(self, parser, namespace, values, option_string=None):
@@ -81,8 +87,10 @@ def check_prompt_field(checker, field = "", password = False):
 	ap.add_argument("-d", "--directory", action=check_dir, default=".", help="directory where the backup file will be stored")
 	ap.add_argument("--http", dest="https", action="store_false", help="use HTTP as protocol")
 	ap.add_argument("--https", dest="https", action="store_true", help="use HTTPS as protocol (default)")
+	ap.add_argument("--ignore-certificate", dest="ignore-cert", action="store_true", help="ignore invalid certificates")
 	ap.add_argument("-v", "--version", action="version", version="%(prog)s 1.2.0")
 	ap.set_defaults(https=True)
+	ap.set_defaults(ssl=False)
 	ap.set_defaults(prompt_pwd=False)
 
 	try:
@@ -108,6 +116,21 @@ def check_prompt_field(checker, field = "", password = False):
 		else:
 			protocol = "http://"
 
+		# Python by default attempts to perform certificate validation since v2.7.9
+		if args["ignore-cert"] and version_info[2] > 8:
+			from functools import wraps
+			from httplib import HTTPSConnection
+			from ssl import _create_unverified_context
+
+			old_init = HTTPSConnection.__init__
+
+			@wraps(HTTPSConnection.__init__)
+			def ignore_cert(self, *args, **kwargs):
+				kwargs["context"] = _create_unverified_context()
+				old_init(self, *args, **kwargs)
+
+			HTTPSConnection.__init__ = ignore_cert
+
 		directory = args["directory"]
 
 		br = Browser()
@@ -144,3 +167,7 @@ def check_prompt_field(checker, field = "", password = False):
 	except ArgumentTypeError as e:
 		stderr.write("\n{0}\n".format(e))
 		exit(1)
+	except CertificateError as e:
+		stderr.write("\n{0}\n".format(e))
+		stderr.write("You can use the '--ignore-certificate' option to ignore this error\n")
+		exit(1)