diff --git a/.pylintrc b/.pylintrc index cd5b093..de9f331 100644 --- a/.pylintrc +++ b/.pylintrc @@ -1,5 +1,5 @@ [BASIC] -good-names=a,b,c,d,df,i,logger,n,tz +good-names=a,b,c,d,df,i,logger,n,on,tz [MASTER] ignore=ci,docs diff --git a/setup.py b/setup.py index 4dda6b1..5e4d2e2 100644 --- a/setup.py +++ b/setup.py @@ -56,7 +56,7 @@ MONGO_REQUIRES = ("pymongo",) -MSSQL_REQUIRES = ("pymssql<3.0", "sqlalchemy") +MSSQL_REQUIRES = ("cython", "pymssql<3.0", "sqlalchemy") KEYWORDS = ( # eg: 'keyword1', 'keyword2', 'keyword3', diff --git a/src/dsdk/mongo.py b/src/dsdk/mongo.py index 2dd3c36..05653ca 100644 --- a/src/dsdk/mongo.py +++ b/src/dsdk/mongo.py @@ -110,11 +110,10 @@ def open_batch( with self.open_mongo() as database: update_one(database.batches, key, doc) - def store_evidence( - self, batch: Batch, *args, exclude: Sequence[str] = () - ) -> None: + def store_evidence(self, batch: Batch, *args, **kwargs) -> None: """Store Evidence.""" - super().store_evidence(batch, *args, exclude) + super().store_evidence(batch, *args, **kwargs) + exclude = kwargs.get("exclude", ()) while args: key, df, *args = args # type: ignore # TODO We need to check column types and convert as needed diff --git a/src/dsdk/service.py b/src/dsdk/service.py index cc5b263..72ba818 100644 --- a/src/dsdk/service.py +++ b/src/dsdk/service.py @@ -156,7 +156,7 @@ def open_batch( # pylint: disable=no-self-use,unused-argument record.end = datetime.now(timezone.utc) def store_evidence( # pylint: disable=no-self-use,unused-argument - self, batch: Batch, *args, exclude: Sequence[str] = () + self, batch: Batch, *args, **kwargs ) -> None: """Store evidence.""" while args: