Unverified Commit 2e575a98 authored by Peter Stanko's avatar Peter Stanko
Browse files

Init data fix

parent 40a4fd97
Loading
Loading
Loading
Loading
Loading
+3 −9
Original line number Diff line number Diff line
@@ -41,15 +41,9 @@ class DataManagement(object):
        Args:
            env(str): Name of the environment
        """
        env = env
        if env == 'prod':
            init_prod_data(db=self.db, app=self.app)
        if env == 'test':
            init_test_data(db=self.db, app=self.app)
        if env == 'cpp':
            init_cpp_data(app=self.app, db=self.db)
        else:
            init_dev_data(db=self.db, app=self.app)
        call = dict(prod=init_prod_data, test=init_test_data, cpp=init_cpp_data, dev=init_dev_data)
        func = call.get(env) or init_dev_data
        func(db=self.db, app=self.app)

    def delete_user(self, name: str):
        """Deletes user