From 28592b4cbfda15f8594b2765d031c777d31cf767 Mon Sep 17 00:00:00 2001 From: Simeon Simeonov Date: Sat, 13 May 2017 16:49:21 +0200 Subject: Make beinc_server.py beinc_poller.py beinc_weechat.py both Python2 and Python>=3.4 compatible --- beinc_poller.py | 18 ++++++++++++++---- beinc_server.py | 20 +++++++++++++++----- beinc_weechat.py | 10 ++++++++-- 3 files changed, 37 insertions(+), 11 deletions(-) diff --git a/beinc_poller.py b/beinc_poller.py index e84ad71..b4e798f 100755 --- a/beinc_poller.py +++ b/beinc_poller.py @@ -1,4 +1,4 @@ -#!/usr/bin/env python3 +#!/usr/bin/env python # -*- coding: utf-8 -*- # Blackmore's Enhanced IRC-Notification Collection (BEINC) v3.0 @@ -29,11 +29,21 @@ import ssl import sys import time -from urllib.parse import urlencode -from urllib.request import urlopen +PY2 = sys.version_info[0] == 2 +PY3 = sys.version_info[0] == 3 + +if PY3: + from urllib.parse import urlencode + from urllib.request import urlopen +else: + from urllib import urlencode + from urllib2 import urlopen try: - import notify2 as pynotify + if PY3: + import notify2 as pynotify + else: + import pynotify except ImportError as e: pynotify = None diff --git a/beinc_server.py b/beinc_server.py index 113856f..3d9ff35 100755 --- a/beinc_server.py +++ b/beinc_server.py @@ -1,4 +1,4 @@ -#!/usr/bin/env python3 +#!/usr/bin/env python # -*- coding: utf-8 -*- # Blackmore's Enhanced IRC-Notification Collection (BEINC) v3.0 @@ -22,7 +22,6 @@ import argparse import cgi import errno import getpass -import http.server import json import logging import os @@ -32,8 +31,19 @@ import sys from functools import wraps from logging.config import fileConfig +PY2 = sys.version_info[0] == 2 +PY3 = sys.version_info[0] == 3 + +if PY3: + from http.server import BaseHTTPRequestHandler, HTTPServer +else: + from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer + try: - import notify2 as pynotify + if PY3: + import notify2 as pynotify + else: + import pynotify except ImportError as e: pynotify = None @@ -182,7 +192,7 @@ class BEINCInstance(object): self.__message_queue.append({'title': title, 'message': message}) -class BEINCCustomHandler(http.server.BaseHTTPRequestHandler): +class BEINCCustomHandler(BaseHTTPRequestHandler): """ """ def do_POST(self): @@ -288,7 +298,7 @@ class BEINCCustomHandler(http.server.BaseHTTPRequestHandler): indent=4).encode('utf-8')) -class BEINCNotifyServer(http.server.HTTPServer): +class BEINCNotifyServer(HTTPServer): """ """ def set_config(self, config): diff --git a/beinc_weechat.py b/beinc_weechat.py index 23d4db8..982e8f0 100644 --- a/beinc_weechat.py +++ b/beinc_weechat.py @@ -24,9 +24,15 @@ import socket import ssl import sys -from urllib import urlencode -from urllib2 import urlopen +PY2 = sys.version_info[0] == 2 +PY3 = sys.version_info[0] == 3 +if PY3: + from urllib.parse import urlencode + from urllib.request import urlopen +else: + from urllib import urlencode + from urllib2 import urlopen import weechat -- cgit v1.3