masonite-framework-orm 3.0.1__py3-none-any.whl

masoniteorm/scopes/__init__.py

@@ -0,0 +1,8 @@
+from .BaseScope import BaseScope
+from .scope import scope
+from .SoftDeleteScope import SoftDeleteScope
+from .SoftDeletesMixin import SoftDeletesMixin
+from .TimeStampsMixin import TimeStampsMixin
+from .TimeStampsScope import TimeStampsScope
+from .UUIDPrimaryKeyMixin import UUIDPrimaryKeyMixin
+from .UUIDPrimaryKeyScope import UUIDPrimaryKeyScope

masoniteorm/scopes/scope.py

@@ -0,0 +1,15 @@
+class scope:
+    def __init__(self, callback, *params, **kwargs):
+        self.fn = callback
+
+    def __set_name__(self, cls, name):
+        if cls not in cls._scopes:
+            cls._scopes[cls] = {name: self.fn}
+        else:
+            cls._scopes[cls].update({name: self.fn})
+        self.cls = cls
+
+    def __call__(self, *args, **kwargs):
+        instantiated = self.cls()
+        builder = instantiated.get_builder()
+        return self.fn(instantiated, builder, *args, **kwargs)

masoniteorm/seeds/Seeder.py

@@ -0,0 +1,42 @@
+import pydoc
+
+
+class Seeder:
+    def __init__(
+        self, dry=False, seed_path="databases/seeds", connection=None
+    ):
+        self.ran_seeds = []
+        self.dry = dry
+        self.seed_path = seed_path
+        self.connection = connection
+        self.seed_module = seed_path.replace("/", ".").replace("\\", ".")
+
+    def call(self, *seeder_classes):
+        for seeder_class in seeder_classes:
+            self.ran_seeds.append(seeder_class)
+            if not self.dry:
+                seeder_class(connection=self.connection).run()
+
+    def run_database_seed(self):
+        database_seeder = pydoc.locate(
+            f"{self.seed_module}.database_seeder.DatabaseSeeder"
+        )
+
+        self.ran_seeds.append(database_seeder)
+
+        if not self.dry:
+            database_seeder(connection=self.connection).run()
+
+    def run_specific_seed(self, seed):
+        file_name = f"{self.seed_module}.{seed}"
+        database_seeder = pydoc.locate(file_name)
+
+        if not database_seeder:
+            raise ValueError(f"Could not find the {file_name} seeder file")
+
+        self.ran_seeds.append(database_seeder)
+
+        if not self.dry:
+            database_seeder(connection=self.connection).run()
+        else:
+            print(f"Running {database_seeder}")

masoniteorm/seeds/__init__.py

@@ -0,0 +1 @@
+from .Seeder import Seeder