chore: run black and clean up imports

Author: Borketh

timetable_kit/amtrak/access.py

@@ -8,8 +8,8 @@
 This has very similar code to baggage.py
 """
 
-from io import StringIO  # for parsing JSON
 import json
+from io import StringIO  # for parsing JSON
 
 import pandas as pd
 

timetable_kit/amtrak/accessibility_check.py

@@ -11,22 +11,18 @@
 Requires local bad_stations.csv file: That is generated by "json_stations.py process"
 """
 import json
-from pathlib import Path
 from io import StringIO  # Needed to parse JSON
 
 import pandas as pd
 
-# These are mine
-
-from timetable_kit.file_locations import get_timetable_kit_data_home
-
 from timetable_kit.amtrak.json_stations import (
     load_stations_json,
     load_station_details,
 )
 
 # To filter out that which is not a train station
 from timetable_kit.amtrak.station_type import is_train_station
+from timetable_kit.file_locations import get_timetable_kit_data_home
 
 station_stats_dir = get_timetable_kit_data_home() / "amtrak" / "station_stats"
 stations_csv_path = get_timetable_kit_data_home() / "amtrak" / "stations.csv"

timetable_kit/amtrak/agency.py

@@ -5,35 +5,33 @@
 
 This holds a class for "AgencyAmtrak" intended to be used as a singleton.
 """
-from timetable_kit.feed_enhanced import FeedEnhanced  # Mostly for typechecking
-
-from timetable_kit.generic_agency import Agency
-
-# for patch_feed
-import timetable_kit.amtrak.gtfs_patches as gtfs_patches
-
 # for patch_add_wheelchair_boarding
 import timetable_kit.amtrak.access as access
 
-# for sleeper trains, which trains have checked baggage, major stations, etc
-import timetable_kit.amtrak.special_data as special_data
-
 # for whether stations have checked baggage
 import timetable_kit.amtrak.baggage as baggage
 
+# for patch_feed
+import timetable_kit.amtrak.gtfs_patches as gtfs_patches
+
 # for get_station_name
 import timetable_kit.amtrak.json_stations as json_stations
 
+# for get_route_name
+import timetable_kit.amtrak.route_names as route_names
+
+# for sleeper trains, which trains have checked baggage, major stations, etc
+import timetable_kit.amtrak.special_data as special_data
+
 # for get_station_name_pretty (subroutines)
 import timetable_kit.text_assembly as text_assembly
-from timetable_kit.text_assembly import SAFE_BR
 
 # Map from station codes to connecting service names
 # This is stashed in a class variable
 from timetable_kit.amtrak.connecting_services_data import connecting_services_dict
-
-# for get_route_name
-import timetable_kit.amtrak.route_names as route_names
+from timetable_kit.feed_enhanced import FeedEnhanced  # Mostly for typechecking
+from timetable_kit.generic_agency import Agency
+from timetable_kit.text_assembly import SAFE_BR
 
 
 class AgencyAmtrak(Agency):

timetable_kit/amtrak/agency_cleanup.py

@@ -71,9 +71,9 @@ def revised_amtrak_agencies(agency):
     )
     # Edit the lookup table:
     # This was only needed for pre-2022 Amtrak data.
-    agency_lookup_table[
-        174
-    ] = "Amtrak Directly Operated Thruway Bus"  # Is "Amtrak" in feed
+    agency_lookup_table[174] = (
+        "Amtrak Directly Operated Thruway Bus"  # Is "Amtrak" in feed
+    )
     agency_lookup_table[192] = "Thruway Bus Operator 192"
     agency_lookup_table[1206] = "Thruway Bus Operator 1206"
     agency_lookup_table[1207] = "Thruway Bus Operator 1207"

timetable_kit/amtrak/baggage.py

@@ -7,8 +7,8 @@
 Requires local copy of Amtrak stations database: That local copy is generated by "json_stations.py download"
 """
 
-from io import StringIO  # for parsing JSON
 import json
+from io import StringIO  # for parsing JSON
 
 import pandas as pd
 

timetable_kit/amtrak/get_wiki_stations.py

@@ -11,13 +11,12 @@
 """
 
 import argparse
-from pathlib import Path
 import re
 from math import nan
+from pathlib import Path
 
 import pandas as pd
 
-
 arg_parser = argparse.ArgumentParser(
     formatter_class=argparse.RawDescriptionHelpFormatter,
     description="""Process the list of Amtrak stations in Wikipedia.

timetable_kit/amtrak/gtfs_patches.py

@@ -7,11 +7,10 @@
 """
 # TODO: all the Amtrak-specific stuff needs to be made object oriented in an "Amtrak object" perhaps
 
-from timetable_kit.debug import debug_print
-from timetable_kit.feed_enhanced import FeedEnhanced
-
 # Add the wheelchair boarding information from JSON into the GTFS
 from timetable_kit.amtrak.access import patch_add_wheelchair_boarding
+from timetable_kit.debug import debug_print
+from timetable_kit.feed_enhanced import FeedEnhanced
 
 arizona_stops_list = [
     # Sunset Limited
@@ -116,7 +115,7 @@ def patch_hiawatha(feed: FeedEnhanced):
 
 
 def patch_sunset_limited(feed: FeedEnhanced):
-    """Patch a bug where Sunset Limited #2 has wrong departure days. 
+    """Patch a bug where Sunset Limited #2 has wrong departure days.
     It departs from LAX on Su/We/Fr and not Sa/Tu/Th
 
     The bug is because it starts so late it starts on the following day in Eastern Standard Time.
@@ -139,17 +138,17 @@ def patch_sunset_limited(feed: FeedEnhanced):
                 debug_print(1, "Found #2 listed as running on Saturday, patching")
                 new_calendar.loc[index, "sunday"] = 0
                 new_calendar.loc[index, "monday"] = 1
-                #feed.calendar = new_calendar
+                # feed.calendar = new_calendar
             if new_calendar.loc[index, "wednesday"] == 1:  # This is incorrrect.
                 debug_print(1, "Found #2 listed as running on Tuesday, patching")
                 new_calendar.loc[index, "wednesday"] = 0
                 new_calendar.loc[index, "thursday"] = 1
-                #feed.calendar = new_calendar
+                # feed.calendar = new_calendar
             if new_calendar.loc[index, "friday"] == 1:  # This is incorrrect.
                 debug_print(1, "Found #2 listed as running on Thursday, patching")
                 new_calendar.loc[index, "friday"] = 0
                 new_calendar.loc[index, "saturday"] = 1
-                #feed.calendar = new_calendar
+                # feed.calendar = new_calendar
             feed.calendar = new_calendar
 
 

timetable_kit/amtrak/json_stations.py

@@ -18,18 +18,14 @@
 # will download Amtrak's station files into the './stations/' directory
 # otherwise runs test case
 
-import sys
-from pathlib import Path
-from io import StringIO  # Needed to parse JSON
 import argparse
-
-import requests
-import json  # better for the details import
+import random
+from io import StringIO  # Needed to parse JSON
+from pathlib import Path
 from time import sleep  # Avoid slamming Amtrak's server too fast -- not needed
 
 import pandas as pd
-
-import random
+import requests
 
 from timetable_kit.file_locations import get_timetable_kit_data_home
 

timetable_kit/amtrak/station_type.py

@@ -13,18 +13,18 @@
 
 import sys  # for sys.exit
 from io import StringIO  # Needed to parse JSON
-import pandas as pd
 
 # For parsing the HTML pages
 import lxml.html
+import pandas as pd
 
 from timetable_kit.amtrak.json_stations import (
     load_stations_json,
     load_station_details_html,
 )
 
 # These are mine
-from timetable_kit.debug import set_debug_level, debug_print
+from timetable_kit.debug import set_debug_level
 
 # This is a map from what we might see in the web page,
 # to the key information in the form:

timetable_kit/amtrak/wiki_station_cleanup.py

@@ -9,9 +9,8 @@
 get station names from Amtrak's JSON data.
 """
 
-import pandas as pd
 import gtfs_kit  # type: ignore # Tell MyPy this has no type stubs
-
+import pandas as pd
 
 # Cities with multiple stations in the same city, requiring disambiguation
 two_station_cities = [