22
33import json
44import logging
5- from typing import Dict , List , Optional , Set , Tuple , Type
5+ from typing import Optional
66
77from grimp .application .ports .filesystem import AbstractFileSystem
88from grimp .application .ports .modulefinder import FoundPackage , ModuleFile
1313from grimp import _rustgrimp as rust # type: ignore[attr-defined]
1414
1515logger = logging .getLogger (__name__ )
16- PrimitiveFormat = Dict [str , List [ Tuple [str , Optional [int ], str ]]]
16+ PrimitiveFormat = dict [str , list [ tuple [str , Optional [int ], str ]]]
1717
1818
1919class CacheFileNamer :
@@ -24,7 +24,7 @@ def make_meta_file_name(cls, found_package: FoundPackage) -> str:
2424 @classmethod
2525 def make_data_file_name (
2626 cls ,
27- found_packages : Set [FoundPackage ],
27+ found_packages : set [FoundPackage ],
2828 include_external_packages : bool ,
2929 exclude_type_checking_imports : bool ,
3030 ) -> str :
@@ -42,7 +42,7 @@ def make_data_file_name(
4242 @classmethod
4343 def make_data_file_unique_string (
4444 cls ,
45- found_packages : Set [FoundPackage ],
45+ found_packages : set [FoundPackage ],
4646 include_external_packages : bool ,
4747 exclude_type_checking_imports : bool ,
4848 ) -> str :
@@ -65,24 +65,24 @@ def make_data_file_unique_string(
6565class Cache (AbstractCache ):
6666 DEFAULT_CACHE_DIR = ".grimp_cache"
6767
68- def __init__ (self , * args , namer : Type [CacheFileNamer ], ** kwargs ) -> None :
68+ def __init__ (self , * args , namer : type [CacheFileNamer ], ** kwargs ) -> None :
6969 """
7070 Don't instantiate Cache directly; use Cache.setup().
7171 """
7272 super ().__init__ (* args , ** kwargs )
73- self ._mtime_map : Dict [str , float ] = {}
74- self ._data_map : Dict [Module , Set [DirectImport ]] = {}
73+ self ._mtime_map : dict [str , float ] = {}
74+ self ._data_map : dict [Module , set [DirectImport ]] = {}
7575 self ._namer = namer
7676
7777 @classmethod
7878 def setup (
7979 cls ,
8080 file_system : AbstractFileSystem ,
81- found_packages : Set [FoundPackage ],
81+ found_packages : set [FoundPackage ],
8282 include_external_packages : bool ,
8383 exclude_type_checking_imports : bool = False ,
84- cache_dir : Optional [ str ] = None ,
85- namer : Type [CacheFileNamer ] = CacheFileNamer ,
84+ cache_dir : str | None = None ,
85+ namer : type [CacheFileNamer ] = CacheFileNamer ,
8686 ) -> "Cache" :
8787 cache = cls (
8888 file_system = file_system ,
@@ -98,10 +98,10 @@ def setup(
9898 return cache
9999
100100 @classmethod
101- def cache_dir_or_default (cls , cache_dir : Optional [ str ] ) -> str :
101+ def cache_dir_or_default (cls , cache_dir : str | None ) -> str :
102102 return cache_dir or cls .DEFAULT_CACHE_DIR
103103
104- def read_imports (self , module_file : ModuleFile ) -> Set [DirectImport ]:
104+ def read_imports (self , module_file : ModuleFile ) -> set [DirectImport ]:
105105 try :
106106 cached_mtime = self ._mtime_map [module_file .module .name ]
107107 except KeyError :
@@ -118,7 +118,7 @@ def read_imports(self, module_file: ModuleFile) -> Set[DirectImport]:
118118
119119 def write (
120120 self ,
121- imports_by_module : Dict [Module , Set [DirectImport ]],
121+ imports_by_module : dict [Module , set [DirectImport ]],
122122 ) -> None :
123123 self ._write_marker_files_if_not_already_there ()
124124 # Write data file.
@@ -165,13 +165,13 @@ def write(
165165 def _build_mtime_map (self ) -> None :
166166 self ._mtime_map = self ._read_mtime_map_files ()
167167
168- def _read_mtime_map_files (self ) -> Dict [str , float ]:
169- all_mtimes : Dict [str , float ] = {}
168+ def _read_mtime_map_files (self ) -> dict [str , float ]:
169+ all_mtimes : dict [str , float ] = {}
170170 for found_package in self .found_packages :
171171 all_mtimes .update (self ._read_mtime_map_file (found_package ))
172172 return all_mtimes
173173
174- def _read_mtime_map_file (self , found_package : FoundPackage ) -> Dict [str , float ]:
174+ def _read_mtime_map_file (self , found_package : FoundPackage ) -> dict [str , float ]:
175175 meta_cache_filename = self .file_system .join (
176176 self .cache_dir , self ._namer .make_meta_file_name (found_package )
177177 )
@@ -191,7 +191,7 @@ def _read_mtime_map_file(self, found_package: FoundPackage) -> Dict[str, float]:
191191 def _build_data_map (self ) -> None :
192192 self ._data_map = self ._read_data_map_file ()
193193
194- def _read_data_map_file (self ) -> Dict [Module , Set [DirectImport ]]:
194+ def _read_data_map_file (self ) -> dict [Module , set [DirectImport ]]:
195195 data_cache_filename = self .file_system .join (
196196 self .cache_dir ,
197197 self ._namer .make_data_file_name (
0 commit comments