Linter hints fixes

This commit is contained in:
M1ha 2021-07-07 17:56:40 +05:00
parent 243333df2a
commit e16892cb3b
12 changed files with 24 additions and 27 deletions

View File

@ -29,6 +29,7 @@ jobs:
--health-retries 5 --health-retries 5
ports: ports:
- 5432:5432 - 5432:5432
clickhouse: clickhouse:
image: yandex/clickhouse-server:${{ matrix.clickhouse-version }} image: yandex/clickhouse-server:${{ matrix.clickhouse-version }}
ports: ports:

View File

@ -94,7 +94,7 @@ class ClickHouseModel(with_metaclass(ClickHouseModelMeta, InfiModel)):
return namedtuple("%sTuple" % cls.__name__, field_names, defaults=default_values) return namedtuple("%sTuple" % cls.__name__, field_names, defaults=default_values)
@classmethod @classmethod
def objects_in(cls, database: Database)-> QuerySet: def objects_in(cls, database: Database) -> QuerySet:
return QuerySet(cls, database) return QuerySet(cls, database)
@classmethod @classmethod
@ -121,7 +121,7 @@ class ClickHouseModel(with_metaclass(ClickHouseModelMeta, InfiModel)):
return connections[db_alias] return connections[db_alias]
@classmethod @classmethod
def get_django_model_serializer(cls, writable: bool= False, defaults: Optional[dict] = None def get_django_model_serializer(cls, writable: bool = False, defaults: Optional[dict] = None
) -> Django2ClickHouseModelSerializer: ) -> Django2ClickHouseModelSerializer:
serializer_cls = lazy_class_import(cls.django_model_serializer) serializer_cls = lazy_class_import(cls.django_model_serializer)
return serializer_cls(cls, writable=writable, defaults=defaults) return serializer_cls(cls, writable=writable, defaults=defaults)

View File

@ -28,7 +28,7 @@ def django_pg_returning_available(using: str) -> bool:
:return: Boolean :return: Boolean
""" """
try: try:
import django_pg_returning import django_pg_returning # noqa: F401
return connections[using].vendor == 'postgresql' return connections[using].vendor == 'postgresql'
except ImportError: except ImportError:
return False return False

View File

@ -9,7 +9,7 @@ from django.db.models.signals import post_migrate
from django.dispatch import receiver from django.dispatch import receiver
# In order to support all operations import here # In order to support all operations import here
from infi.clickhouse_orm.migrations import * from infi.clickhouse_orm.migrations import * # noqa F401, F403
from infi.clickhouse_orm.database import ServerError, DatabaseException from infi.clickhouse_orm.database import ServerError, DatabaseException
from infi.clickhouse_orm.fields import StringField, DateField from infi.clickhouse_orm.fields import StringField, DateField

View File

@ -83,7 +83,7 @@ class Storage:
:param kwargs: Storage dependant arguments :param kwargs: Storage dependant arguments
:return: Number of records in queue :return: Number of records in queue
""" """
raise NotImplemented() raise NotImplementedError()
def get_operations(self, import_key: str, count: int, **kwargs) -> List[Tuple[str, str]]: def get_operations(self, import_key: str, count: int, **kwargs) -> List[Tuple[str, str]]:
""" """
@ -94,7 +94,7 @@ class Storage:
:param kwargs: Storage dependant arguments :param kwargs: Storage dependant arguments
:return: A list of tuples (operation, pk) in incoming order. :return: A list of tuples (operation, pk) in incoming order.
""" """
raise NotImplemented() raise NotImplementedError()
def register_operations(self, import_key: str, operation: str, *pks: Any) -> int: def register_operations(self, import_key: str, operation: str, *pks: Any) -> int:
""" """
@ -135,21 +135,21 @@ class Storage:
This method is used in tests to drop all storage data This method is used in tests to drop all storage data
:return: None :return: None
""" """
raise NotImplemented() raise NotImplementedError()
def get_last_sync_time(self, import_key: str) -> Optional[datetime.datetime]: def get_last_sync_time(self, import_key: str) -> Optional[datetime.datetime]:
""" """
Gets the last time, sync has been executed Gets the last time, sync has been executed
:return: datetime.datetime if last sync has been. Otherwise - None. :return: datetime.datetime if last sync has been. Otherwise - None.
""" """
raise NotImplemented() raise NotImplementedError()
def set_last_sync_time(self, import_key: str, dt: datetime.datetime) -> None: def set_last_sync_time(self, import_key: str, dt: datetime.datetime) -> None:
""" """
Sets successful sync time Sets successful sync time
:return: None :return: None
""" """
raise NotImplemented() raise NotImplementedError()
class RedisStorage(with_metaclass(SingletonMeta, Storage)): class RedisStorage(with_metaclass(SingletonMeta, Storage)):

View File

@ -15,9 +15,9 @@ os.environ.setdefault("DJANGO_SETTINGS_MODULE", 'tests.settings')
django.setup() django.setup()
# This imports must be after django activation # This imports must be after django activation
from django.db.models import F from django.db.models import F # noqa: E402
from tests.clickhouse_models import ClickHouseCollapseTestModel from tests.clickhouse_models import ClickHouseCollapseTestModel # noqa: E402
from tests.models import TestModel from tests.models import TestModel # noqa: E402
logger = logging.getLogger('django-clickhouse') logger = logging.getLogger('django-clickhouse')

View File

@ -48,5 +48,3 @@ class NamedTupleTest(TestCase):
t2 = TestTuple(1, 2, 3) t2 = TestTuple(1, 2, 3)
self.assertEqual(t1, t2) self.assertEqual(t1, t2)
self.assertEqual((1, 2, 3), t1) self.assertEqual((1, 2, 3), t1)

View File

@ -2,7 +2,7 @@ import datetime
from unittest import skipIf from unittest import skipIf
import django import django
from django.test import TransactionTestCase, TestCase from django.test import TransactionTestCase
from django.utils.timezone import now from django.utils.timezone import now
from tests.clickhouse_models import ClickHouseTestModel, ClickHouseSecondTestModel, ClickHouseCollapseTestModel, \ from tests.clickhouse_models import ClickHouseTestModel, ClickHouseSecondTestModel, ClickHouseCollapseTestModel, \

View File

@ -75,11 +75,10 @@ class StorageTest(TestCase):
def test_locks(self): def test_locks(self):
# Test that multiple can acquire locks in parallel # Test that multiple can acquire locks in parallel
# And single model can't # And single model can't
l = self.storage.get_lock(ClickHouseTestModel.get_import_key()) lock = self.storage.get_lock(ClickHouseTestModel.get_import_key())
l.acquire() lock.acquire()
with self.assertRaises(RedisLockTimeoutError): with self.assertRaises(RedisLockTimeoutError):
l.acquire() lock.acquire()
l2 = self.storage.get_lock(ClickHouseCollapseTestModel.get_import_key())
l2.acquire()
lock_2 = self.storage.get_lock(ClickHouseCollapseTestModel.get_import_key())
lock_2.acquire()

View File

@ -119,4 +119,3 @@ class TestSingletonMeta(TestCase):
b = Single() b = Single()
self.assertEqual(a, b) self.assertEqual(a, b)
self.assertEqual(2, b.test) self.assertEqual(2, b.test)