🪓 fix mypy issues

Author: shroominic

.github/workflows/pre-commit.yml

@@ -9,7 +9,7 @@ jobs:
     - uses: actions/checkout@v2
     - uses: actions/setup-python@v2
       with:
-        python-version: 3.8
+        python-version: 3.9
     - name: Install pre-commit
       run: pip install pre-commit
     - name: Run pre-commit

codeboxapi/box/basebox.py

@@ -3,7 +3,7 @@
 from abc import ABC, abstractmethod
 from datetime import datetime
 from os import PathLike
-from typing import Optional
+from typing import List, Optional
 from uuid import UUID
 
 from typing_extensions import Self
@@ -76,11 +76,11 @@ async def ainstall(self, package_name: str) -> CodeBoxStatus:
         """Async Install a python package to the venv"""
 
     @abstractmethod
-    def list_files(self) -> list[CodeBoxFile]:
+    def list_files(self) -> List[CodeBoxFile]:
         """List all available files inside the CodeBox instance"""
 
     @abstractmethod
-    async def alist_files(self) -> list[CodeBoxFile]:
+    async def alist_files(self) -> List[CodeBoxFile]:
         """Async List all available files inside the CodeBox instance"""
 
     @abstractmethod

codeboxapi/box/codebox.py

@@ -34,7 +34,7 @@
 """
 
 from os import PathLike
-from typing import Any, Optional
+from typing import Any, Dict, List, Optional
 from uuid import UUID
 
 from aiohttp import ClientSession
@@ -62,7 +62,7 @@ def __init__(self, *args, **kwargs) -> None:
         self.session_id: Optional[UUID] = kwargs.get("id", None)
         self.aiohttp_session: Optional[ClientSession] = None
 
-    def codebox_request(self, method, endpoint, *args, **kwargs) -> dict[str, Any]:
+    def codebox_request(self, method, endpoint, *args, **kwargs) -> Dict[str, Any]:
         """Basic request to the CodeBox API"""
         self._update()
         return base_request(
@@ -71,7 +71,7 @@ def codebox_request(self, method, endpoint, *args, **kwargs) -> dict[str, Any]:
 
     async def acodebox_request(
         self, method, endpoint, *args, **kwargs
-    ) -> dict[str, Any]:
+    ) -> Dict[str, Any]:
         """Basic async request to the CodeBox API"""
         self._update()
         if self.aiohttp_session is None:
@@ -213,7 +213,7 @@ async def ainstall(self, package_name: str) -> CodeBoxStatus:
             )
         )
 
-    def list_files(self) -> list[CodeBoxFile]:
+    def list_files(self) -> List[CodeBoxFile]:
         return [
             CodeBoxFile(name=file_name, content=None)
             for file_name in (
@@ -224,7 +224,7 @@ def list_files(self) -> list[CodeBoxFile]:
             )["files"]
         ]
 
-    async def alist_files(self) -> list[CodeBoxFile]:
+    async def alist_files(self) -> List[CodeBoxFile]:
         return [
             CodeBoxFile(name=file_name, content=None)
             for file_name in (

codeboxapi/box/localbox.py

@@ -11,7 +11,7 @@
 import subprocess
 import time
 from asyncio.subprocess import Process
-from typing import Optional, Union
+from typing import List, Optional, Union
 from uuid import uuid4
 
 import aiohttp
@@ -449,13 +449,13 @@ async def ainstall(self, package_name: str) -> CodeBoxStatus:
         # restart kernel if needed TODO
         return CodeBoxStatus(status=f"{package_name} installed successfully")
 
-    def list_files(self) -> list[CodeBoxFile]:
+    def list_files(self) -> List[CodeBoxFile]:
         return [
             CodeBoxFile(name=file_name, content=None)
             for file_name in os.listdir(".codebox")
         ]
 
-    async def alist_files(self) -> list[CodeBoxFile]:
+    async def alist_files(self) -> List[CodeBoxFile]:
         return await asyncio.to_thread(self.list_files)
 
     def restart(self) -> CodeBoxStatus: