Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions cachier/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,10 +21,10 @@
from typing import TYPE_CHECKING, Callable, Literal, Optional, TypedDict, Union
from warnings import warn

from .base_core import RecalculationNeeded, _BaseCore
from .memory_core import _MemoryCore
from .mongo_core import _MongoCore
from .pickle_core import _PickleCore
from .cores.base import RecalculationNeeded, _BaseCore
from .cores.memory import _MemoryCore
from .cores.mongo import _MongoCore
from .cores.pickle import _PickleCore

if TYPE_CHECKING:
import pymongo.collection
Expand Down
Empty file added cachier/cores/__init__.py
Empty file.
File renamed without changes.
2 changes: 1 addition & 1 deletion cachier/memory_core.py → cachier/cores/memory.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import threading
from datetime import datetime

from .base_core import _BaseCore
from .base import _BaseCore


class _MemoryCore(_BaseCore):
Expand Down
2 changes: 1 addition & 1 deletion cachier/mongo_core.py → cachier/cores/mongo.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from pymongo.errors import OperationFailure
from bson.binary import Binary # to save binary data to mongodb

from .base_core import _BaseCore, RecalculationNeeded
from .base import _BaseCore, RecalculationNeeded


MONGO_SLEEP_DURATION_IN_SEC = 1
Expand Down
2 changes: 1 addition & 1 deletion cachier/pickle_core.py → cachier/cores/pickle.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@

# Alternative: https://github.com/WoLpH/portalocker

from .base_core import _BaseCore
from .base import _BaseCore


class _PickleCore(_BaseCore):
Expand Down
3 changes: 2 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""Setup file for the Cachier package."""
from setuptools.config.expand import find_packages

# This file is part of Cachier.
# https://github.com/shaypal5/cachier
Expand Down Expand Up @@ -30,7 +31,7 @@
author='Shay Palachy',
author_email='shay.palachy@gmail.com',
url='https://github.com/python-cachier/cachier',
packages=['cachier', 'cachier.scripts'],
packages=find_packages(exclude=['tests']),
entry_points='''
[console_scripts]
cachier=cachier.scripts.cli:cli
Expand Down
10 changes: 5 additions & 5 deletions tests/test_mongo_core.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,8 @@
from pymongo_inmemory import MongoClient as InMemoryMongoClient

from cachier import cachier
from cachier.base_core import RecalculationNeeded
from cachier.mongo_core import _MongoCore
from cachier.cores.base import RecalculationNeeded
from cachier.cores.mongo import _MongoCore


# === Enables testing vs a real MongoDB instance ===
Expand Down Expand Up @@ -270,9 +270,9 @@ def mock_get_entry_by_key(self, key): # skipcq: PYL-R0201, PYL-W0613
return "key", None

monkeypatch.setattr(
"cachier.mongo_core._MongoCore.get_entry", mock_get_entry)
"cachier.cores.mongo._MongoCore.get_entry", mock_get_entry)
monkeypatch.setattr(
"cachier.mongo_core._MongoCore.get_entry_by_key", mock_get_entry_by_key
"cachier.cores.mongo._MongoCore.get_entry_by_key", mock_get_entry_by_key
)

@cachier(mongetter=_test_mongetter)
Expand All @@ -290,7 +290,7 @@ def mock_get_entry_2(self, args, kwargs): # skipcq: PYL-W0613
return "key", entry

monkeypatch.setattr(
"cachier.mongo_core._MongoCore.get_entry", mock_get_entry_2)
"cachier.cores.mongo._MongoCore.get_entry", mock_get_entry_2)

stale_after = datetime.timedelta(seconds=1)

Expand Down