odie/colorspot
1
0
Fork 0
mirror of https://codeberg.org/mclemens/colorspot.git synced 2024-06-03 03:30:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

moved all files to ~/.config/colorspot/

Browse Source
This commit is contained in:
Michael Clemens 2022-05-20 18:05:10 +02:00
parent c54b7d4c11
commit 585153e84b
3 changed files with 26 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
setup.cfg
View File

@ -1,6 +1,6 @@
[metadata] [metadata]
name = colorspot name = colorspot
version = 0.1.1 version = 0.1.3
author = Michael Clemens author = Michael Clemens
author_email = colorspot@qrz.is author_email = colorspot@qrz.is
description = A colorful CLI DX cluster client with LotW integration description = A colorful CLI DX cluster client with LotW integration
@ -23,7 +23,6 @@ python_requires = >=3.6
install_requires= install_requires=
colored colored
requests requests
zipfile
[options.packages.find] [options.packages.find]
where = src where = src

2
src/colorspot/__init__.py
View File

@ -1,3 +1,3 @@
from colorspot.__main__ import ColorSpot from colorspot.__main__ import ColorSpot
__version__ = '0.1.1' __version__ = '0.1.3'

41
src/colorspot/__main__.py
View File

@ -37,6 +37,7 @@ import zipfile
from telnetlib import Telnet from telnetlib import Telnet
import requests import requests
from colored import fg, bg, attr from colored import fg, bg, attr
from pathlib import Path
class ColorSpot(): class ColorSpot():
"""ColorSpot class""" """ColorSpot class"""
@ -47,7 +48,11 @@ class ColorSpot():
self.print_banner() self.print_banner()
self.config = configparser.ConfigParser() self.config = configparser.ConfigParser()
self.config_file = os.path.expanduser('~/.colorspot.ini') self.home_dir = str(Path.home())
self.config_dir = self.home_dir + "/.config/colorspot/"
# Check if config directory exists and else create it
Path(self.config_dir).mkdir(parents=True, exist_ok=True)
self.config_file = os.path.expanduser(self.config_dir + 'colorspot.ini')
self.read_config(self.config, self.config_file) self.read_config(self.config, self.config_file)
self.check_files() self.check_files()
@ -56,7 +61,7 @@ class ColorSpot():
self.confirmed_entities = self.get_confirmed_entities() self.confirmed_entities = self.get_confirmed_entities()
if self.check_cty: if self.check_cty:
with open(self.config['files']['cty'], encoding='us-ascii') as csvfile: with open(self.config_dir + self.config['files']['cty'], encoding='us-ascii') as csvfile:
self.cty = list(csv.reader(csvfile, delimiter=',')) self.cty = list(csv.reader(csvfile, delimiter=','))
@ -156,10 +161,11 @@ class ColorSpot():
def check_files(self): def check_files(self):
"""Checks if all necessary files are in the file system. """Checks if all necessary files are in the file system.
Downloads all files and unzips them (if necessary)""" Downloads all files and unzips them (if necessary)"""
# check for lotw qsl information file # check for lotw qsl information file
self.check_lotw_confirmed = exists(self.config['files']['lotw_confirmed']) self.check_lotw_confirmed = exists(self.config_dir + self.config['files']['lotw_confirmed'])
if not self.check_lotw_confirmed: if not self.check_lotw_confirmed:
print("The file " + self.config['files']['lotw_confirmed'] + " is missing.") print("The file " + self.config_dir + self.config['files']['lotw_confirmed'] + " is missing.")
user = self.config['lotw']['user'] user = self.config['lotw']['user']
password = self.config['lotw']['password'] password = self.config['lotw']['password']
mode = self.config['lotw']['mode'] mode = self.config['lotw']['mode']
@ -167,37 +173,38 @@ class ColorSpot():
"&qso_query=1&qso_qsl=yes&qso_mode={}&qso_qsldetail=yes&"\ "&qso_query=1&qso_qsl=yes&qso_mode={}&qso_qsldetail=yes&"\
"qso_qslsince=1970-01-01".format(user, password, mode) "qso_qslsince=1970-01-01".format(user, password, mode)
print("Trying to download " + url) print("Trying to download " + url)
self.download_file(url, self.config['files']['lotw_confirmed']) self.download_file(url, self.config_dir + self.config['files']['lotw_confirmed'])
self.check_lotw_confirmed = exists(self.config['files']['lotw_confirmed']) self.check_lotw_confirmed = exists(self.config_dir + self.config['files']['lotw_confirmed'])
if self.check_lotw_confirmed: if self.check_lotw_confirmed:
print("File successfully downloaded") print("File successfully downloaded")
else: else:
print("something went wrong while downloading " + url) print("something went wrong while downloading " + url)
# check for cty.csv file # check for cty.csv file
self.check_cty = exists(self.config['files']['cty']) self.check_cty = exists(self.config_dir + self.config['files']['cty'])
if not self.check_cty: if not self.check_cty:
url = self.config['files']['cty_url'] url = self.config['files']['cty_url']
print("The file " + self.config['files']['cty'] + " is missing.") print("The file " + self.config_dir + self.config['files']['cty'] + " is missing.")
print("Trying to download " + url) print("Trying to download " + url)
zip_name = self.download_file(url, "bigcty.zip" ) # TODO: pfad?
zip_name = self.download_file(url, self.config_dir + "bigcty.zip" )
with zipfile.ZipFile(zip_name, 'r') as zip_ref: with zipfile.ZipFile(zip_name, 'r') as zip_ref:
zip_ref.extract("cty.csv") zip_ref.extract("cty.csv", path=self.config_dir)
os.remove(zip_name) os.remove(zip_name)
self.check_cty = exists(self.config['files']['cty']) self.check_cty = exists(self.config_dir + self.config['files']['cty'])
if self.check_cty: if self.check_cty:
print("File successfully downloaded and extracted.") print("File successfully downloaded and extracted.")
else: else:
print("something went wrong while downloading " + url) print("something went wrong while downloading " + url)
# check for lotw user activity file # check for lotw user activity file
self.check_lotw_activity = exists(self.config['files']['lotw_activity']) self.check_lotw_activity = exists(self.config_dir + self.config['files']['lotw_activity'])
if not self.check_lotw_activity: if not self.check_lotw_activity:
url = self.config['files']['lotw_activity_url'] url = self.config['files']['lotw_activity_url']
print("The file " + self.config['files']['lotw_activity'] + " is missing.") print("The file " + self.config_dir + self.config['files']['lotw_activity'] + " is missing.")
print("Trying to download " + url) print("Trying to download " + url)
self.download_file(url, self.config['files']['lotw_activity']) self.download_file(url, self.config_dir + self.config['files']['lotw_activity'])
self.check_lotw_activity = exists(self.config['files']['lotw_activity']) self.check_lotw_activity = exists(self.config_dir + self.config['files']['lotw_activity'])
if self.check_lotw_activity: if self.check_lotw_activity:
print("File successfully downloaded") print("File successfully downloaded")
else: else:
@ -208,7 +215,7 @@ class ColorSpot():
"""Reads the file downlaoded from LotW with all confirmed QSOs, """Reads the file downlaoded from LotW with all confirmed QSOs,
extracts all confirmed DXCCs and puts them into a list""" extracts all confirmed DXCCs and puts them into a list"""
ret = [] ret = []
with open(self.config['files']['lotw_confirmed'], encoding='us-ascii') as file: with open(self.config_dir + self.config['files']['lotw_confirmed'], encoding='us-ascii') as file:
for row in file: for row in file:
if re.search("<DXCC:", row): if re.search("<DXCC:", row):
dxcc = row.partition(">")[2].lower().rstrip() dxcc = row.partition(">")[2].lower().rstrip()
@ -221,7 +228,7 @@ class ColorSpot():
"""Reads the LotW user activity file and returns the date """Reads the LotW user activity file and returns the date
of the last upload date if a specific call sign""" of the last upload date if a specific call sign"""
ret = "" ret = ""
with open(self.config['files']['lotw_activity'], encoding='us-ascii') as csvfile: with open(self.config_dir + self.config['files']['lotw_activity'], encoding='us-ascii') as csvfile:
csv_file = csv.reader(csvfile, delimiter=',') csv_file = csv.reader(csvfile, delimiter=',')
#loop through the csv file #loop through the csv file
for row in csv_file: for row in csv_file: