first commit

colorspot.py

@@ -0,0 +1,161 @@
+import sys
+import csv
+import re
+import json
+import os
+from telnetlib import Telnet
+from colored import fg, bg, attr
+import configparser
+
+class ColorSpot():
+
+    def __init__(self):
+        self.version = "0.1.0"
+
+        self.config = configparser.ConfigParser()
+        self.config_file = os.path.expanduser('~/.colorspot.ini')
+        self.read_config(self.config, self.config_file)
+
+        self.country_data = {}
+        #TODO: download file and/or tell user what to do
+        with open('cty.json', "r") as json_file:
+            self.country_data = json.load(json_file)
+
+        self.qsl_countries = self.get_confirmed_countries()
+
+    @staticmethod
+    def read_config(config, file_name):
+        """reads the configuration from the config file or
+        creates a default config file if none could be found"""
+        if os.path.isfile(file_name):
+            config.read(file_name)
+        else:
+            config = configparser.ConfigParser()
+            config['cluster'] = {
+                'host': 'dxc.nc7j.com',
+                'port': '7373',
+                'timeout': '100'}
+            config['colors'] = {
+                'use_colors': 'yes',
+                "145": "white",
+                "144": "white",
+                "50": "white",
+                "29": "yellow",
+                "28": "yellow",
+                "24": "red",
+                "21": "orchid",
+                "18": "green",
+                "14": "steel_blue_3",
+                "10": "orange_1",
+                "7": "cyan",
+                "5": "white",
+                "3": "light_cyan",
+                "1": "white",
+                'alert_bg': 'indian_red_1a',
+                'alert_fg': 'white',
+                'default_bg': 'black'}
+            with open(file_name, 'w') as configfile:
+                config.write(configfile)
+            print("\nNo configuration file found. A new configuration file has been created.")
+            print("\nPlease edit the file " + file_name + " and restart the application.\n" )
+            sys.exit()
+        return config
+
+    @staticmethod
+    def get_confirmed_countries():
+        ret = []
+        #TODO: download file and/or tell user what to do
+        file = open("lotwreport.adi", "r")
+        for row in file:
+            if re.search("COUNTRY", row):
+                country = row.partition(">")[2].lower().rstrip()
+                if country not in ret:
+                    ret.append(country)
+        return ret
+
+    @staticmethod
+    def check_lotw(call):
+        ret = ""
+        #TODO: download file and/or tell user what to do
+        csv_file = csv.reader(open('lotw-user-activity.csv', "r"), delimiter=",")
+        #loop through the csv file
+        for row in csv_file:
+            if call == row[0]:
+                ret = row[1]
+                return ret
+        return ret
+
+    @staticmethod
+    def get_country(call, data):
+        done = False
+        while not done:
+            for number in data:
+                for country in data[number]:
+                    if call == country["prefix"]:
+                        return country["areaname"]
+            call = call[:-1]
+        return None
+
+    def get_spots(self):
+        with Telnet(self.config['cluster']['host'], int(self.config['cluster']['port']), \
+                int(self.config['cluster']['timeout'])) as telnet:
+            while True:
+                line_enc = telnet.read_until(b"\n")  # Read one line
+                line = line_enc.decode('ascii')
+                if "enter your call" in line:
+                    my_str_as_bytes = str.encode(self.config['cluster']['user']+"\n")
+                    telnet.write(my_str_as_bytes)
+                elif "DX de" in line:
+                    try:
+                        band_col = ""
+                        call_de = re.search('DX de (.+?): ', line).group(1)
+                        freq = re.search(': +(.+?)  ', line).group(1)
+                        call_dx = re.search(freq + ' +(.+?) ', line).group(1)
+                        time = re.search('[^ ]*$', line).group(0)[0:4]
+                        comment = re.search(call_dx + ' +(.+?) +' + time, line).group(1)
+                        areaname = self.get_country(call_dx, self.country_data)
+                        lotw_date = self.check_lotw(call_dx)
+
+                        lotw = ""
+                        if lotw_date:
+                            lotw = "[LotW] "
+
+                        try:
+                            band_col = self.config['colors'][freq[:-5]]
+                        except Exception:
+                            band_col = "white"
+
+                        freq = freq.replace('.0', '')
+
+                        if areaname.lower() not in self.qsl_countries:
+                            background = self.config['colors']['alert_bg']
+                            band_col = self.config['colors']['alert_fg']
+                        else:
+                            background = self.config['colors']['default_bg']
+
+                        sep = fg("grey_27")+'|'+fg(band_col)
+
+                        row = [call_de, sep, freq, sep, call_dx, \
+                            sep, areaname, sep, lotw + comment, sep, time]
+
+                        print(bg(background) + fg(band_col) + \
+                            '{:9.9} {:<1} {:>7.7} {:<1} {:<10.10} {:<1} {:<16.16} {:<1} {:<30.30} {:<1} {:<4.4}'.format(*row) + attr("reset"))
+
+                    except AttributeError:
+                        print(line)
+                elif "login: " in line:
+                    print(line)
+
+#####################################################
+#                  Main Routine                     #
+#####################################################
+def main():
+    """the main routine"""
+    color_spot = ColorSpot()
+    color_spot.get_spots()
+
+if __name__ == "__main__":
+    try:
+        sys.exit(main())
+    except EOFError:
+        pass