diff --git a/arib/ass.py b/arib/ass.py index fee4be6..297ac4b 100644 --- a/arib/ass.py +++ b/arib/ass.py @@ -12,18 +12,19 @@ file. """ -from pathlib import Path -from enum import Enum import copy +import re +import unicodedata from dataclasses import dataclass +from enum import Enum +from pathlib import Path from typing import List -import unicodedata + import arib.code_set as code_set import arib.control_characters as control_characters -import re from arib.arib_exceptions import FileOpenError -from arib.drcs_decoder import drcs_unpack_to_bitmap from arib.drcs_cache import DrcsGlyph +from arib.drcs_decoder import drcs_unpack_to_bitmap # DRCS drawing support diff --git a/arib/autosub.py b/arib/autosub.py index e8af0a7..e08319b 100755 --- a/arib/autosub.py +++ b/arib/autosub.py @@ -10,22 +10,19 @@ embedded in MPEG Transport Stream data. """ -from arib.bing import translate -import arib.code_set as code_set -import arib.control_characters as control_characters +import argparse import codecs +import copy +import os import re -import os -import argparse -import copy -from arib.closed_caption import next_data_unit -from arib.closed_caption import StatementBody -from arib.mpeg.ts import next_ts_packet -from arib.mpeg.ts import PESPacket +import arib.code_set as code_set +import arib.control_characters as control_characters +from arib.bing import translate +from arib.closed_caption import StatementBody, next_data_unit from arib.data_group import DataGroup -from arib.secret_key import SECRET_KEY -from arib.secret_key import CLIENT_ID +from arib.mpeg.ts import PESPacket, next_ts_packet +from arib.secret_key import CLIENT_ID, SECRET_KEY class Pos(object): diff --git a/arib/bing.py b/arib/bing.py index dcea9b6..3ad311e 100755 --- a/arib/bing.py +++ b/arib/bing.py @@ -12,9 +12,10 @@ """ import argparse import json -import requests import urllib +import requests + def translate(text, from_language="ja", to_language="en", client_id="", secret_key=""): if not secret_key: diff --git a/arib/closed_caption.py b/arib/closed_caption.py index 5e05af2..36d63e6 100644 --- a/arib/closed_caption.py +++ b/arib/closed_caption.py @@ -14,12 +14,14 @@ from arib import read from arib.decoder import Decoder -from arib.drcs_cache import DRCS_CACHE -from arib.drcs_cache import DrcsGlyph -from arib.drcs_cache import drcs_set_from_font_id_byte -from arib.drcs_cache import normalize_94 -from arib.drcs_cache import is_94_byte -from arib.drcs_cache import drcs0_pack +from arib.drcs_cache import ( + DRCS_CACHE, + DrcsGlyph, + drcs0_pack, + drcs_set_from_font_id_byte, + is_94_byte, + normalize_94, +) DEBUG = False DRCS_DEBUG = False diff --git a/arib/code_set.py b/arib/code_set.py index 969d4eb..f3555ce 100644 --- a/arib/code_set.py +++ b/arib/code_set.py @@ -11,11 +11,9 @@ """ -from arib.arib_exceptions import UnimplimentedError from arib import read -from arib.drcs_cache import DRCS_CACHE -from arib.drcs_cache import normalize_94 -from arib.drcs_cache import drcs0_pack +from arib.arib_exceptions import UnimplimentedError +from arib.drcs_cache import DRCS_CACHE, drcs0_pack, normalize_94 DEBUG = False diff --git a/arib/control_characters.py b/arib/control_characters.py index e0fc320..a178106 100644 --- a/arib/control_characters.py +++ b/arib/control_characters.py @@ -9,9 +9,8 @@ """ from arib import read -from arib.code_set import code_set_handler_from_final_byte -from arib.code_set import in_code_set_table from arib.arib_exceptions import DecodingError +from arib.code_set import code_set_handler_from_final_byte, in_code_set_table DEBUG = False diff --git a/arib/data_group.py b/arib/data_group.py index 9053b82..bee8570 100644 --- a/arib/data_group.py +++ b/arib/data_group.py @@ -11,12 +11,11 @@ """ import sys -from arib import read -from arib.read import EOFError import traceback -from arib.closed_caption import CaptionStatementData -from arib.closed_caption import CaptionManagementData +from arib import read +from arib.closed_caption import CaptionManagementData, CaptionStatementData +from arib.read import EOFError DEBUG = False diff --git a/arib/decoder.py b/arib/decoder.py index 48172f1..18f7131 100644 --- a/arib/decoder.py +++ b/arib/decoder.py @@ -8,12 +8,10 @@ """ -from arib import read -from arib.control_characters import is_control_character -from arib.control_characters import handle_control_character import arib.control_characters as control_char -from arib import code_set +from arib import code_set, read from arib.arib_exceptions import DecodingError +from arib.control_characters import handle_control_character, is_control_character DEBUG = False diff --git a/arib/drcs_cache.py b/arib/drcs_cache.py index 07a5c86..cff7c82 100644 --- a/arib/drcs_cache.py +++ b/arib/drcs_cache.py @@ -1,7 +1,8 @@ -from dataclasses import dataclass -from typing import Tuple, Optional -from collections import OrderedDict import threading +from collections import OrderedDict +from dataclasses import dataclass +from typing import Optional, Tuple + from arib import read diff --git a/arib/es_extract.py b/arib/es_extract.py index 814cb6a..ae235fa 100755 --- a/arib/es_extract.py +++ b/arib/es_extract.py @@ -7,21 +7,17 @@ DATE: Saturday January 14th 2017 """ -import sys import argparse +import sys import traceback - from pathlib import Path - -from arib.closed_caption import next_data_unit -from arib.closed_caption import StatementBody import arib.code_set as code_set import arib.control_characters as control_characters -from arib.data_group import next_data_group # print out some additional info for DRCS values -from arib.closed_caption import set_DRCS_debug +from arib.closed_caption import StatementBody, next_data_unit, set_DRCS_debug +from arib.data_group import next_data_group set_DRCS_debug(True) diff --git a/arib/mpeg/ts.py b/arib/mpeg/ts.py index 10a63b1..5c16714 100755 --- a/arib/mpeg/ts.py +++ b/arib/mpeg/ts.py @@ -7,13 +7,13 @@ DATE: Thursday, October 20th 2016 """ -import os -import sys import argparse -import struct # memorymap file on 64 bit systems import mmap +import os +import struct +import sys class ES: diff --git a/arib/srt.py b/arib/srt.py index 48d3634..2d059cb 100644 --- a/arib/srt.py +++ b/arib/srt.py @@ -1,9 +1,9 @@ -from pathlib import Path -from enum import Enum -import sys import math -from typing import Optional, Union, Callable, Dict, Any import re +import sys +from enum import Enum +from pathlib import Path +from typing import Any, Callable, Dict, Optional, Union import arib.code_set as code_set import arib.control_characters as control_characters diff --git a/arib/ts2ass.py b/arib/ts2ass.py index 7a664ca..5cfd44c 100755 --- a/arib/ts2ass.py +++ b/arib/ts2ass.py @@ -11,24 +11,19 @@ from __future__ import annotations -import sys import argparse +import sys import traceback from dataclasses import dataclass from pathlib import Path from typing import Optional -from arib.read import EOFError - -from arib.closed_caption import next_data_unit -from arib.closed_caption import StatementBody -from arib.data_group import DataGroup from arib.arib_exceptions import FileOpenError - -from arib.mpeg.ts import TS -from arib.mpeg.ts import ES - from arib.ass import ASSFormatter +from arib.closed_caption import StatementBody, next_data_unit +from arib.data_group import DataGroup +from arib.mpeg.ts import ES, TS +from arib.read import EOFError @dataclass(frozen=True) diff --git a/arib/ts2srt.py b/arib/ts2srt.py index 7ddbe53..df080ab 100755 --- a/arib/ts2srt.py +++ b/arib/ts2srt.py @@ -9,23 +9,18 @@ from __future__ import annotations -import sys import argparse +import sys import traceback from dataclasses import dataclass from pathlib import Path from typing import Optional -from arib.read import EOFError - -from arib.closed_caption import next_data_unit -from arib.closed_caption import StatementBody -from arib.data_group import DataGroup from arib.arib_exceptions import FileOpenError - -from arib.mpeg.ts import TS -from arib.mpeg.ts import ES - +from arib.closed_caption import StatementBody, next_data_unit +from arib.data_group import DataGroup +from arib.mpeg.ts import ES, TS +from arib.read import EOFError from arib.srt import SRTFormatter @@ -128,7 +123,7 @@ def on_es_packet(self, current_pid, packet, header_size): "Exception thrown while handling DataGroup in ES." "This may be due to many factors " "such as file corruption or the .ts file using" - "as yet unsupported features." + " as yet unsupported features." ) traceback.print_exc(file=sys.stdout) diff --git a/arib/ts_extract.py b/arib/ts_extract.py index 84b42b1..a06094d 100755 --- a/arib/ts_extract.py +++ b/arib/ts_extract.py @@ -7,23 +7,19 @@ DATE: Thursday, October 20th 2016 """ +import argparse import os import sys -import argparse import traceback -from arib.read import EOFError - -from arib.mpeg.ts import TS -from arib.mpeg.ts import ES -from arib.closed_caption import next_data_unit -from arib.closed_caption import StatementBody import arib.code_set as code_set import arib.control_characters as control_characters -from arib.data_group import DataGroup # print out some additional info for DRCS values -from arib.closed_caption import set_DRCS_debug +from arib.closed_caption import StatementBody, next_data_unit, set_DRCS_debug +from arib.data_group import DataGroup +from arib.mpeg.ts import ES, TS +from arib.read import EOFError set_DRCS_debug(True) diff --git a/pyproject.toml b/pyproject.toml index 9bc6fa7..8ff1fdf 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -61,5 +61,5 @@ line-length = 100 target-version = "py310" [tool.ruff.lint] -select = ["E", "F", "W"] # pycodestyle/pyflakes/warnings +select = ["E", "F", "W", "I"] # pycodestyle/pyflakes/warnings ignore = [] \ No newline at end of file diff --git a/tests/test_cli_es_extract.py b/tests/test_cli_es_extract.py index f3864d5..151bcc3 100644 --- a/tests/test_cli_es_extract.py +++ b/tests/test_cli_es_extract.py @@ -1,5 +1,7 @@ import sys + import pytest + import arib.es_extract as cli # List of input .es files to run the CLI against diff --git a/tests/test_read.py b/tests/test_read.py index 7f1a7e8..3fb990f 100644 --- a/tests/test_read.py +++ b/tests/test_read.py @@ -1,4 +1,5 @@ import io + import arib.read as read