ENH: Add python loggers

CHANGELOG.md

@@ -39,6 +39,7 @@ Attention: The newest changes should be on top -->
 - ENH: Auto Populate Changelog [#919](https://github.com/RocketPy-Team/RocketPy/pull/919)
 - ENH: Adaptive Monte Carlo via Convergence Criteria [#922](https://github.com/RocketPy-Team/RocketPy/pull/922)
 - TST: Add acceptance tests for 3DOF flight simulation based on Bella Lui rocket [#914](https://github.com/RocketPy-Team/RocketPy/pull/914)
+- ENH: adopt built-in Python logging instead of print() calls [#450](https://github.com/RocketPy-Team/RocketPy/issues/450)
 
 ### Changed
 

rocketpy/__init__.py

@@ -1,3 +1,5 @@
+import logging
+
 from .control import _Controller
 from .environment import Environment, EnvironmentAnalysis
 from .mathutils import (
@@ -47,6 +49,7 @@
     TrapezoidalFins,
 )
 from .sensitivity import SensitivityModel
+from . import utils
 from .sensors import Accelerometer, Barometer, GnssReceiver, Gyroscope
 from .simulation import Flight, MonteCarlo, MultivariateRejectionSampler
 from .stochastic import (
@@ -62,3 +65,5 @@
     StochasticTail,
     StochasticTrapezoidalFins,
 )
+
+logging.getLogger(__name__).addHandler(logging.NullHandler())

rocketpy/environment/environment.py

@@ -1,8 +1,10 @@
 # pylint: disable=too-many-public-methods, too-many-instance-attributes
 import bisect
 import json
+import logging
 import re
 import warnings
+
 from collections import namedtuple
 from datetime import datetime
 
@@ -47,6 +49,8 @@
     geopotential_height_to_geometric_height,
 )
 
+logger = logging.getLogger(__name__)
+
 
 class Environment:
     """Keeps all environment information stored, such as wind and temperature
@@ -977,7 +981,7 @@ def set_elevation(self, elevation="Open-Elevation"):
             self.elevation = elevation
         else:
             self.elevation = fetch_open_elevation(self.latitude, self.longitude)
-            print(f"Elevation received: {self.elevation} m")
+            logger.info("Elevation received: %.2f m", self.elevation)
 
     def set_topographic_profile(  # pylint: disable=redefined-builtin, unused-argument
         self, type, file, dictionary="netCDF4", crs=None
@@ -1016,14 +1020,13 @@ def set_topographic_profile(  # pylint: disable=redefined-builtin, unused-argume
                 # crsArray = nasa_dem.variables['crs'][:].tolist().
                 self.topographic_profile_activated = True
 
-                print("Region covered by the Topographical file: ")
-                print(
-                    f"Latitude from {self.elev_lat_array[-1]:.6f}° to "
-                    f"{self.elev_lat_array[0]:.6f}°"
-                )
-                print(
-                    f"Longitude from {self.elev_lon_array[0]:.6f}° to "
-                    f"{self.elev_lon_array[-1]:.6f}°"
+                logger.debug(
+                    "Topographical file coverage: lat [%.6f°, %.6f°], "
+                    "lon [%.6f°, %.6f°]",
+                    self.elev_lat_array[-1],
+                    self.elev_lat_array[0],
+                    self.elev_lon_array[0],
+                    self.elev_lon_array[-1],
                 )
 
     def get_elevation_from_topographic_profile(self, lat, lon):
@@ -2738,9 +2741,10 @@ def export_environment(self, filename="environment"):
 
         with open(filename + ".json", "w") as f:
             json.dump(export_env_dictionary, f, sort_keys=False, indent=4, default=str)
-        print(
-            f"Your Environment file was saved at '{filename}.json'. You can use "
-            "it in the future by using the custom_atmosphere atmospheric model."
+        logger.info(
+            "Your Environment file was saved at '%s.json'. "
+            "You can use it in the future by using the custom_atmosphere atmospheric model.",
+            filename,
         )
 
     def set_earth_geometry(self, datum):
@@ -2973,6 +2977,6 @@ def from_dict(cls, data):  # pylint: disable=too-many-statements
 
     results = doctest.testmod()
     if results.failed < 1:
-        print(f"All the {results.attempted} tests passed!")
+        logger.debug("All the %d tests passed!", results.attempted)
     else:
-        print(f"{results.failed} out of {results.attempted} tests failed.")
+        logger.error("%d out of %d tests failed.", results.failed, results.attempted)

rocketpy/environment/environment_analysis.py

@@ -2,7 +2,9 @@
 import copy
 import datetime
 import json
+import logging
 import warnings
+
 from collections import defaultdict
 from functools import cached_property
 
@@ -24,6 +26,8 @@
 from ..units import convert_units
 from .environment import Environment
 
+logger = logging.getLogger(__name__)
+
 # TODO: the average_wind_speed_profile_by_hour and similar methods could be more abstract than currently are
 
 
@@ -236,13 +240,15 @@ def __check_requirements(self):
                 check_requirement_version(module_name, version)
             except (ValueError, ImportError) as e:
                 has_error = True
-                print(
-                    f"The following error occurred while importing {module_name}: {e}"
+                logger.error(
+                    "The following error occurred while importing %s: %s",
+                    module_name,
+                    e,
                 )
         if has_error:
-            print(
+            logger.error(
                 "Given the above errors, some methods may not work. Please run "
-                + "'pip install rocketpy[env_analysis]' to install extra requirements."
+                "'pip install rocketpy[env_analysis]' to install extra requirements."
             )
 
     def __init_surface_dictionary(self):
@@ -508,8 +514,9 @@ def __init_unit_system(self):
             }
         else:
             # Default to SI
-            print(
-                f"Defaulting to SI unit system, the {self.unit_system_string} was not found."
+            logger.warning(
+                "Defaulting to SI unit system, the '%s' unit system was not found.",
+                self.unit_system_string,
             )
             self.unit_system = {
                 "length": "m",
@@ -2843,9 +2850,10 @@ def export_mean_profiles(self, filename="export_env_analysis"):
                     self.export_dictionary, sort_keys=False, indent=4, default=str
                 )
             )
-        print(
-            f"Your Environment Analysis file was saved, check it out: {filename}.json\n"
-            "You can use it to set a `customAtmosphere` atmospheric model"
+        logger.info(
+            "Your Environment Analysis file was saved: %s.json. "
+            "You can use it to set a customAtmosphere atmospheric model.",
+            filename,
         )
 
     @classmethod
@@ -2884,7 +2892,7 @@ def save(self, filename="env_analysis_dict"):
         file = open(filename, "w")  # pylint: disable=consider-using-with
         file.write(encoded_class)
         file.close()
-        print("Your Environment Analysis file was saved, check it out: " + filename)
+        logger.info("Your Environment Analysis file was saved: %s", filename)
 
     def create_environment_object(
         self, gravity=None, date=None, datum="SIRGAS2000", max_expected_height=80000.0

rocketpy/environment/fetchers.py

@@ -3,6 +3,7 @@
 functions may be changed without notice in future feature releases.
 """
 
+import logging
 import re
 import time
 from datetime import datetime, timedelta, timezone
@@ -12,6 +13,8 @@
 
 from rocketpy.tools import exponential_backoff
 
+logger = logging.getLogger(__name__)
+
 MAX_RETRY_DELAY_SECONDS = 600
 
 
@@ -37,7 +40,9 @@ def fetch_open_elevation(lat, lon):
     RuntimeError
         If there is a problem reaching the Open-Elevation API servers.
     """
-    print(f"Fetching elevation from open-elevation.com for lat={lat}, lon={lon}...")
+    logger.debug(
+        "Fetching elevation from open-elevation.com for lat=%s, lon=%s", lat, lon
+    )
     request_url = f"https://api.open-elevation.com/api/v1/lookup?locations={lat},{lon}"
     try:
         response = requests.get(request_url)

rocketpy/environment/tools.py

@@ -5,14 +5,18 @@
 future to improve their performance and usability.
 """
 
+import logging
 import math
+
 import warnings
 
 import netCDF4
 import numpy as np
 
 from rocketpy.tools import bilinear_interpolation
 
+logger = logging.getLogger(__name__)
+
 ## Wind data functions
 
 
@@ -850,6 +854,6 @@ def utm_to_geodesic(  # pylint: disable=too-many-locals,too-many-statements
 
     results = doctest.testmod()
     if results.failed < 1:
-        print(f"All the {results.attempted} tests passed!")
+        logger.info("All the %d tests passed!", results.attempted)
     else:
-        print(f"{results.failed} out of {results.attempted} tests failed.")
+        logger.error("%d out of %d tests failed.", results.failed, results.attempted)

rocketpy/mathutils/function.py

@@ -5,8 +5,10 @@
 carefully as it may impact all the rest of the project.
 """
 
+import logging
 import operator
 import warnings
+
 from bisect import bisect_left
 from collections.abc import Iterable
 from copy import deepcopy
@@ -30,6 +32,8 @@
 from rocketpy.plots.plot_helpers import show_or_save_plot
 from rocketpy.tools import deprecated, from_hex_decode, to_hex_encode
 
+logger = logging.getLogger(__name__)
+
 NUMERICAL_TYPES = (float, int, complex, np.integer, np.floating)
 INTERPOLATION_TYPES = {
     "linear": 0,
@@ -2018,7 +2022,7 @@ def plot(self, *args, **kwargs):
             elif self.__dom_dim__ == 2:
                 self.plot_2d(*args, **kwargs)
             else:
-                print("Error: Only functions with 1D or 2D domains can be plotted.")
+                logger.error("Only functions with 1D or 2D domains can be plotted.")
 
     @deprecated(
         reason="The `Function.plot1D` method is set to be deprecated and fully "
@@ -4405,6 +4409,6 @@ def reset_funcified_methods(instance):
 
     results = doctest.testmod()
     if results.failed < 1:
-        print(f"All the {results.attempted} tests passed!")
+        logger.info("All the %d tests passed!", results.attempted)
     else:
-        print(f"{results.failed} out of {results.attempted} tests failed.")
+        logger.error("%d out of %d tests failed.", results.failed, results.attempted)

rocketpy/mathutils/vector_matrix.py

@@ -1,9 +1,12 @@
+import logging
 from cmath import isclose
 from functools import cached_property
 from itertools import product
 
 from rocketpy.tools import euler313_to_quaternions, normalize_quaternions
 
+logger = logging.getLogger(__name__)
+
 
 class Vector:
     """Pure python basic R3 vector class designed for simple operations.
@@ -1108,6 +1111,6 @@ def from_dict(cls, data):
 
     results = doctest.testmod()
     if results.failed < 1:
-        print(f"All the {results.attempted} tests passed!")
+        logger.info("All the %d tests passed!", results.attempted)
     else:
-        print(f"{results.failed} out of {results.attempted} tests failed.")
+        logger.error("%d out of %d tests failed.", results.failed, results.attempted)

rocketpy/motors/tank_geometry.py

@@ -1,3 +1,4 @@
+import logging
 import warnings
 from functools import cached_property
 
@@ -8,6 +9,8 @@
 from ..plots.tank_geometry_plots import _TankGeometryPlots
 from ..prints.tank_geometry_prints import _TankGeometryPrints
 
+logger = logging.getLogger(__name__)
+
 try:
     from functools import cache
 except ImportError:
@@ -458,11 +461,12 @@ def add_spherical_caps(self):
         part. The height is not modified, meaning that the total volume of
         the tank will decrease.
         """
-        print(
-            "Warning: Adding spherical caps to the tank will not modify the "
-            + f"total height of the tank {self.height} m. "
-            + "Its cylindrical portion height will be reduced to "
-            + f"{self.height - 2 * self.radius_function} m."
+        logger.warning(
+            "Adding spherical caps to the tank will not modify the total height "
+            "of the tank (%.4f m). Its cylindrical portion height will be "
+            "reduced to %s m.",
+            self.height,
+            self.height - 2 * self.radius_function,
         )
 
         if not self.has_caps:

rocketpy/plots/aero_surface_plots.py

@@ -1,13 +1,17 @@
 # pylint: disable=too-many-statements
 
+import logging
 from abc import ABC, abstractmethod
 
 import matplotlib.pyplot as plt
+
 import numpy as np
 from matplotlib.patches import Ellipse
 
 from .plot_helpers import show_or_save_plot
 
+logger = logging.getLogger(__name__)
+
 
 class _AeroSurfacePlots(ABC):
     """Abstract class that contains all aero surface plots."""
@@ -157,7 +161,7 @@ def airfoil(self, *, filename=None):
         """
 
         if self.aero_surface.airfoil:
-            print("Airfoil lift curve:")
+            logger.info("Airfoil lift curve:")
             self.aero_surface.airfoil_cl.plot_1d(force_data=True, filename=filename)
 
     def roll(self, *, filename=None):
@@ -173,7 +177,7 @@ def roll(self, *, filename=None):
         -------
         None
         """
-        print("Roll parameters:")
+        logger.info("Roll parameters:")
         self.aero_surface.roll_parameters[0](filename=filename)
         self.aero_surface.roll_parameters[1](filename=filename)
 
@@ -194,7 +198,7 @@ class for more information on how this plot is made. Also, this method
         -------
         None
         """
-        print("Lift coefficient:")
+        logger.info("Lift coefficient:")
         self.aero_surface.cl(filename=filename)
         self.aero_surface.clalpha_single_fin(filename=filename)
         self.aero_surface.clalpha_multiple_fins(filename=filename)
@@ -238,7 +242,7 @@ def airfoil(self, *, filename=None):
         """
 
         if self.aero_surface.airfoil:
-            print("Airfoil lift curve:")
+            logger.info("Airfoil lift curve:")
             self.aero_surface.airfoil_cl.plot_1d(force_data=True, filename=filename)
 
     def roll(self, *, filename=None):
@@ -254,7 +258,7 @@ def roll(self, *, filename=None):
         -------
         None
         """
-        print("Roll parameters:")
+        logger.info("Roll parameters:")
         self.aero_surface.roll_parameters[0](filename=filename)
         self.aero_surface.roll_parameters[1](filename=filename)
 
@@ -275,7 +279,7 @@ class for more information on how this plot is made. Also, this method
         -------
         None
         """
-        print("Lift coefficient:")
+        logger.info("Lift coefficient:")
         self.aero_surface.cl(filename=filename)
         self.aero_surface.clalpha_single_fin(filename=filename)