Add ruff

Signed-off-by: Daniel Hjelseth Høyer <github@dahoiv.net>
Danielhiversen · Dec 28, 2023 · 6c9b5a3 · 6c9b5a3
1 parent 526f00a
commit 6c9b5a3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 7 deletions.
diff --git a/.github/workflows/code_checker.yml b/.github/workflows/code_checker.yml
@@ -27,7 +27,7 @@ jobs:
       - name: Install depencency
         run: |
             pip install -r requirements.txt
-            pip install dlint black flake8 flake8-bugbear flake8-deprecated flake8-executable isort pylint pytest pytest-cov pytest-asyncio mypy types-python-dateutil ruff
+            pip install dlint black flake8 flake8-bugbear flake8-deprecated flake8-executable isort pylint pytest pytest-cov pytest-asyncio mypy ruff
       - name: Flake8 Code Linter
         run: |
             flake8 $SRC_FOLDER

diff --git a/setup.py b/setup.py
@@ -11,10 +11,7 @@
     packages=["tibber"],
     install_requires=[
         "aiohttp>=3.0.6",
-        "async_timeout>=1.4.0",
         "gql>=3.0.0",
-        "python-dateutil",
-        "websockets>=10.0"
     ],
     package_data={"tibber": ["py.typed"]},
     version=consts["__version__"],

diff --git a/tibber/__init__.py b/tibber/__init__.py
@@ -6,7 +6,6 @@
 from typing import Any
 
 import aiohttp
-import async_timeout
 
 from .const import API_ENDPOINT, DEFAULT_TIMEOUT, DEMO_TOKEN, __version__
 from .exceptions import (
@@ -100,8 +99,7 @@ async def execute(
             "data": payload,
         }
         try:
-            async with async_timeout.timeout(timeout):
-                resp = await self.websession.post(API_ENDPOINT, **post_args)
+            resp = await self.websession.post(API_ENDPOINT, **post_args, timeout=self.timeout)
             return (await extract_response_data(resp)).get("data")
         except (aiohttp.ClientError, asyncio.TimeoutError) as err:
             if retry > 0: