locker: port base implementation from poetry

python-poetry · Sep 22, 2020 · 37c2809 · 37c2809
1 parent 5566f7d
commit 37c2809
Show file tree

Hide file tree

Showing 5 changed files with 829 additions and 2 deletions.
diff --git a/poetry/core/lock/__init__.py b/poetry/core/lock/__init__.py
@@ -0,0 +1,8 @@
+from poetry.core.lock.categories import LockCategory
+from poetry.core.lock.locker import Locker
+
+
+__all__ = [
+    LockCategory.__name__,
+    Locker.__name__,
+]
diff --git a/poetry/core/lock/categories.py b/poetry/core/lock/categories.py
@@ -0,0 +1,11 @@
+from enum import Enum
+
+
+class LockCategory(Enum):
+    MAIN = "main"
+    DEV = "dev"
+
+    def __eq__(self, other):
+        if not isinstance(other, Enum):
+            return self.value == other
+        super(LockCategory, self).__eq__(other)