Merge f1faad1 into 04e7a36

MasoniteFramework · Jul 29, 2018 · a43f9bb · a43f9bb
2 parents 04e7a36 + f1faad1
commit a43f9bb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/masonite/commands/SeedCommand.py b/masonite/commands/SeedCommand.py
@@ -11,7 +11,14 @@ class SeedCommand(Command):
     """
 
     def handle(self):
+        if not 'sys.path.append(os.getcwd())' in open('databases/seeds/__init__.py').read():
+            f = open('databases/seeds/__init__.py', 'w+')
+            f.write('import os\nimport sys\nsys.path.append(os.getcwd())\n')
+            f.close()
+
         table = self.argument('table').lower()
         subprocess.call([
             "orator make:seed {}_table_seeder -p databases/seeds".format(table),
         ], shell=True)
+
+
diff --git a/masonite/commands/SeedRunCommand.py b/masonite/commands/SeedRunCommand.py
@@ -11,6 +11,11 @@ class SeedRunCommand(Command):
     """
 
     def handle(self):
+        if not 'sys.path.append(os.getcwd())' in open('databases/seeds/__init__.py').read():
+            f = open('databases/seeds/__init__.py', 'w+')
+            f.write('import os\nimport sys\nsys.path.append(os.getcwd())\n')
+            f.close()
+
         table = self.argument('table').lower()
         if not table == 'none':
             seeder = '--seeder {}_table_seeder'.format(table.lower())