some linting and readability improvements

This commit is contained in:
simon 2021-08-28 22:51:21 +07:00
parent 84fb419931
commit a5ba8fbbe2
4 changed files with 11 additions and 10 deletions

View File

@ -8,11 +8,11 @@ from time import sleep
from src.config import get_config from src.config import get_config
import src.tvsort as tvsort from src import tvsort
import src.moviesort as moviesort from src import moviesort
import src.db_export as db_export from src import db_export
import src.trailers as trailers from src import trailers
import src.id_fix as id_fix from src import id_fix
class Interface: class Interface:

View File

@ -16,7 +16,7 @@ def get_config():
else: else:
config_path = path.join(root_folder, 'config.json') config_path = path.join(root_folder, 'config.json')
# parse # parse
with open(config_path, 'r') as config_file: with open(config_path, 'r', encoding='utf-8') as config_file:
data = config_file.read() data = config_file.read()
config = json.loads(data) config = json.loads(data)
return config return config

View File

@ -126,7 +126,7 @@ class DatabaseExport:
file_path = path.join(log_folder, filename) file_path = path.join(log_folder, filename)
# open and write # open and write
with open(file_path, 'w') as f: with open(file_path, 'w', encoding='utf-8') as f:
# take fieldnames from first line # take fieldnames from first line
fieldnames = to_write[0].keys() fieldnames = to_write[0].keys()
csv_writer = csv.DictWriter(f, fieldnames) csv_writer = csv.DictWriter(f, fieldnames)
@ -138,7 +138,7 @@ class DatabaseExport:
log_folder = self.CONFIG['media']['log_folder'] log_folder = self.CONFIG['media']['log_folder']
file_path = path.join(log_folder, filename) file_path = path.join(log_folder, filename)
# movie by new # movie by new
with open(file_path, 'w') as f: with open(file_path, 'w', encoding='utf-8') as f:
for line in to_write: for line in to_write:
f.write(line + '\n') f.write(line + '\n')

View File

@ -67,7 +67,7 @@ class TrailerHandler:
""" read log file to get list of trailers to ignore """ """ read log file to get list of trailers to ignore """
log_folder = self.CONFIG['media']['log_folder'] log_folder = self.CONFIG['media']['log_folder']
log_file = os.path.join(log_folder, 'trailers') log_file = os.path.join(log_folder, 'trailers')
with open(log_file, 'r') as log_file: with open(log_file, 'r', encoding='utf-8') as log_file:
trailer_lines = log_file.readlines() trailer_lines = log_file.readlines()
ignore_trailer_list = [] ignore_trailer_list = []
for trailer_line in trailer_lines: for trailer_line in trailer_lines:
@ -97,6 +97,7 @@ class TrailerHandler:
return pending return pending
def dl_pending(self): def dl_pending(self):
# pylint: disable=broad-except
""" download pending trailers """ """ download pending trailers """
sortpath = self.CONFIG['media']['sortpath'] sortpath = self.CONFIG['media']['sortpath']
log_folder = self.CONFIG['media']['log_folder'] log_folder = self.CONFIG['media']['log_folder']
@ -121,7 +122,7 @@ class TrailerHandler:
if i == 4: if i == 4:
# giving up # giving up
log_file = os.path.join(log_folder, 'trailers') log_file = os.path.join(log_folder, 'trailers')
with open(log_file, 'a') as f: with open(log_file, 'a', encoding='utf-8') as f:
f.write(f'{youtube_id} {movie_name}\n') f.write(f'{youtube_id} {movie_name}\n')
break break
sleep((i + 1) ** 2) sleep((i + 1) ** 2)