MNT: Lose some relative imports to avoid bugs

Author: Gui-FernandesBR

rocketpy/plots/__init__.py

@@ -1,18 +0,0 @@
-from .aero_surface_plots import (
-    _EllipticalFinsPlots,
-    _FinsPlots,
-    _NoseConePlots,
-    _TailPlots,
-    _TrapezoidalFinsPlots,
-)
-from .compare import Compare, CompareFlights
-from .environment_analysis_plots import _EnvironmentAnalysisPlots
-from .environment_plots import _EnvironmentPlots
-from .flight_plots import _FlightPlots
-from .hybrid_motor_plots import _HybridMotorPlots
-from .liquid_motor_plots import _LiquidMotorPlots
-from .motor_plots import _MotorPlots
-from .rocket_plots import _RocketPlots
-from .solid_motor_plots import _SolidMotorPlots
-from .tank_geometry_plots import _TankGeometryPlots
-from .tank_plots import _TankPlots

rocketpy/rocket/__init__.py

@@ -1,4 +1,4 @@
-from .aero_surface import (
+from rocketpy.rocket.aero_surface import (
     AeroSurface,
     EllipticalFins,
     Fins,
@@ -7,6 +7,6 @@
     Tail,
     TrapezoidalFins,
 )
-from .components import Components
-from .parachute import Parachute
-from .rocket import Rocket
+from rocketpy.rocket.components import Components
+from rocketpy.rocket.parachute import Parachute
+from rocketpy.rocket.rocket import Rocket

rocketpy/rocket/rocket.py

@@ -2,20 +2,20 @@
 
 import numpy as np
 
-from ..mathutils.function import Function
-from ..motors.motor import EmptyMotor
-from ..plots.rocket_plots import _RocketPlots
-from ..prints.rocket_prints import _RocketPrints
-from .aero_surface import (
+from rocketpy.mathutils.function import Function
+from rocketpy.motors.motor import EmptyMotor
+from rocketpy.plots.rocket_plots import _RocketPlots
+from rocketpy.prints.rocket_prints import _RocketPrints
+from rocketpy.rocket.aero_surface import (
     EllipticalFins,
     Fins,
     NoseCone,
     RailButtons,
     Tail,
     TrapezoidalFins,
 )
-from .components import Components
-from .parachute import Parachute
+from rocketpy.rocket.components import Components
+from rocketpy.rocket.parachute import Parachute
 
 
 class Rocket:
@@ -1183,6 +1183,30 @@ def add_thrust_eccentricity(self, x, y):
         # Return self
         return self
 
+    def draw(self, vis_args=None):
+        """Draws the rocket in a matplotlib figure.
+        
+        Parameters
+        ----------
+        vis_args : dict, optional
+            Determines the visual aspects when drawing the rocket. If None,
+            default values are used. Default values are:
+            {
+                "background": "#EEEEEE",
+                "tail": "black",
+                "nose": "black",
+                "body": "dimgrey",
+                "fins": "black",
+                "motor": "black",
+                "buttons": "black",
+                "line_width": 2.0,
+            }
+            A full list of color names can be found at:
+            https://matplotlib.org/stable/gallery/color/named_colors
+        """
+        self.plots.draw(vis_args)
+        return None
+
     def info(self):
         """Prints out a summary of the data and graphs available about
         the Rocket.