minor fixes

lib/controllers/rocket.py

@@ -1,9 +1,10 @@
-from fastapi import Response, status
 from typing import Dict, Any, Union
+import jsonpickle
+import ast
+
+from fastapi import Response, status
 #from inspect import getsourcelines
 
-import ast
-import jsonpickle
 import rocketpy.Parachute
 import rocketpy.Rocket
 from rocketpy.AeroSurface import NoseCone as rocketpy_NoseCone

lib/models/motor.py

@@ -1,5 +1,5 @@
-from pydantic import BaseModel
 from typing import Optional, Tuple
+from pydantic import BaseModel
 
 class Motor(BaseModel, frozen=True):
     burn_time: float

lib/models/rocket.py

@@ -1,5 +1,5 @@
-from pydantic import BaseModel
 from typing import Optional, Tuple
+from pydantic import BaseModel
 from lib.models.motor import Motor
 from lib.models.aerosurfaces import Fins, NoseCone, Tail, RailButtons
 from lib.models.parachute import Parachute

lib/repositories/environment.py

@@ -56,10 +56,11 @@ def update_env(self) -> "Union[int, None]":
             environment_to_dict = self.environment.dict()
             environment_to_dict["env_id"] = self.environment.__hash__()
 
-            updated_env = self.collection.update_one(
+            self.collection.update_one(
                 { "env_id": self.env_id },
                 { "$set": environment_to_dict }
             )
+
             self.env_id = environment_to_dict["env_id"]
             return self.env_id
         except:
@@ -77,8 +78,7 @@ def get_env(self) -> "Union[Env, None]":
             if environment is not None:
                 del environment["_id"]
                 return Env.parse_obj(environment)
-            else:
-                return None
+            return None
         except:
             raise Exception("Error getting environment")
 

lib/repositories/motor.py

@@ -78,8 +78,7 @@ def get_motor(self) -> "Union[motor, None]":
             if motor is not None:
                 del motor["_id"]
                 return Motor.parse_obj(motor)
-            else:
-                return None
+            return None
         except:
             raise Exception("Error getting motor")
 

lib/repositories/rocket.py

@@ -78,8 +78,7 @@ def get_rocket(self) -> "Union[Rocket, None]":
             if rocket is not None:
                 del rocket["_id"]
                 return Rocket.parse_obj(rocket)
-            else:
-                return None
+            return None
         except:
             raise Exception("Error getting rocket")
 

lib/secrets.py

@@ -15,6 +15,7 @@ def get_os_secret(key):
     def get_secret(self, key):
         dotenv_secret = self.secrets.get(key)
         if not dotenv_secret:
+            print("cheguei")
             return self.get_os_secret(key)
         return dotenv_secret