sapiopycommons 2024.3.19a157__py3-none-any.whl → 2025.1.17a402__py3-none-any.whl

sapiopycommons/general/time_util.py

@@ -1,8 +1,13 @@
+from __future__ import annotations
+
 import time
 from datetime import datetime
+from typing import Any
 
 import pytz
 
+from sapiopycommons.general.exceptions import SapioException
+
 __timezone = None
 """The default timezone. Use TimeUtil.set_default_timezone in a global context before making use of TimeUtil."""
 
@@ -24,7 +29,7 @@ class TimeUtil:
     with static date fields, use "UTC" as your input timezone.
     """
     @staticmethod
-    def get_default_timezone():
+    def get_default_timezone() -> Any:
         """
         Returns the timezone that TimeUtil is currently using as its default.
         """
@@ -43,7 +48,7 @@ class TimeUtil:
         __timezone = TimeUtil.__to_tz(new_timezone)
 
     @staticmethod
-    def __to_tz(timezone: str | int = None):
+    def __to_tz(timezone: str | int = None) -> Any:
         """
         :param timezone: Either the name of a timezone, a UTC offset in seconds, or None if the default should be used.
         :return: The timezone object to use for the given input. If the input is None, uses the default timezone.
@@ -53,11 +58,28 @@ class TimeUtil:
             # because pytz may return timezones from strings in Local Mean Time instead of a timezone with a UTC offset.
             # LMT may be a few minutes off of the actual time in that timezone right now.
             # https://stackoverflow.com/questions/35462876
-            offset: int = int(datetime.now(pytz.timezone(timezone)).utcoffset().total_seconds() / 60)
-            return pytz.FixedOffset(offset)
+            offset: int = TimeUtil.__get_timezone_offset(timezone)
+            # This function takes an offset in minutes, so divide the provided offset seconds by 60.
+            return pytz.FixedOffset(offset // 60)
         if isinstance(timezone, int):
             return pytz.FixedOffset(timezone // 60)
-        return TimeUtil.get_default_timezone()
+        if timezone is None:
+            return TimeUtil.get_default_timezone()
+        raise SapioException(f"Unhandled timezone object of type {type(timezone)}: {timezone}")
+
+    @staticmethod
+    def __get_timezone_offset(timezone: str | int | None) -> int:
+        """
+        :param timezone: Either the name of a timezone, a UTC offset in seconds, or None if the default should be used.
+        :return: The UTC offset in seconds of the provided timezone.
+        """
+        if isinstance(timezone, int):
+            return timezone
+        if isinstance(timezone, str):
+            timezone = pytz.timezone(timezone)
+        if timezone is None:
+            timezone = TimeUtil.get_default_timezone()
+        return int(datetime.now(timezone).utcoffset().total_seconds())
 
     @staticmethod
     def current_time(timezone: str | int = None) -> datetime:
@@ -92,9 +114,10 @@ class TimeUtil:
         return TimeUtil.current_time(timezone).strftime(time_format)
 
     @staticmethod
-    def millis_to_format(millis: int, time_format: str, timezone: str | int = None) -> str:
+    def millis_to_format(millis: int, time_format: str, timezone: str | int = None) -> str | None:
         """
-        Convert the input time in milliseconds to the provided format.
+        Convert the input time in milliseconds to the provided format. If None is passed to the millis parameter,
+        None will be returned
 
         :param millis: The time in milliseconds to convert from.
         :param time_format: The format to display the input time in. Documentation for how the time formatting works
@@ -103,6 +126,9 @@ class TimeUtil:
             timezone variable set by the TimeUtil. A list of valid timezones can be found at
             https://en.wikipedia.org/wiki/List_of_tz_database_time_zones. May also accept a UTC offset in seconds.
         """
+        if millis is None:
+            return None
+
         tz = TimeUtil.__to_tz(timezone)
         return datetime.fromtimestamp(millis / 1000, tz).strftime(time_format)
 
@@ -121,6 +147,64 @@ class TimeUtil:
         tz = TimeUtil.__to_tz(timezone)
         return int(datetime.strptime(time_point, time_format).replace(tzinfo=tz).timestamp() * 1000)
 
+    # FR-47296: Provide functions for shifting between timezones.
+    @staticmethod
+    def shift_now(to_timezone: str = "UTC", from_timezone: str | None = None) -> int:
+        """
+        Take the current time in from_timezone and output the epoch timestamp that would display that same time in
+        to_timezone. A use case for this is when dealing with static date fields to convert a provided timestamp to the
+        value necessary to display that timestamp in the same way when viewed in the static date field.
+
+        :param to_timezone: The timezone to shift to. If not provided, uses UTC.
+        :param from_timezone: The timezone to shift from. If no timezone is provided, uses the global
+            timezone variable set by the TimeUtil. A list of valid timezones can be found at
+            https://en.wikipedia.org/wiki/List_of_tz_database_time_zones. May also accept a UTC offset in seconds.
+        :return: The epoch timestamp that would display as the same time in to_timezone as the current time in
+            from_timezone.
+        """
+        millis: int = TimeUtil.now_in_millis()
+        return TimeUtil.shift_millis(millis, to_timezone, from_timezone)
+
+    @staticmethod
+    def shift_millis(millis: int, to_timezone: str = "UTC", from_timezone: str | None = None) -> int:
+        """
+        Take a number of milliseconds for a time in from_timezone and output the epoch timestamp that would display that
+        same time in to_timezone. A use case for this is when dealing with static date fields to convert a provided
+        timestamp to the value necessary to display that timestamp in the same way when viewed in the static date field.
+
+        :param millis: The time in milliseconds to convert from.
+        :param to_timezone: The timezone to shift to. If not provided, uses UTC.
+        :param from_timezone: The timezone to shift from. If no timezone is provided, uses the global
+            timezone variable set by the TimeUtil. A list of valid timezones can be found at
+            https://en.wikipedia.org/wiki/List_of_tz_database_time_zones. May also accept a UTC offset in seconds.
+        :return: The epoch timestamp that would display as the same time in to_timezone as the given time in
+            from_timezone.
+        """
+        to_offset: int = TimeUtil.__get_timezone_offset(to_timezone) * 1000
+        from_offset: int = TimeUtil.__get_timezone_offset(from_timezone) * 1000
+        return millis + from_offset - to_offset
+
+    @staticmethod
+    def shift_format(time_point: str, time_format: str, to_timezone: str = "UTC", from_timezone: str | None = None) \
+            -> int:
+        """
+        Take a timestamp for a time in from_timezone and output the epoch timestamp that would display that same time
+        in to_timezone. A use case for this is when dealing with static date fields to convert a provided timestamp to
+        the value necessary to display that timestamp in the same way when viewed in the static date field.
+
+        :param time_point: The time in some date/time format to convert from.
+        :param time_format: The format that the time_point is in. Documentation for how the time formatting works
+            can be found at https://docs.python.org/3.10/library/datetime.html#strftime-and-strptime-behavior
+        :param to_timezone: The timezone to shift to. If not provided, uses UTC.
+        :param from_timezone: The timezone to shift from. If no timezone is provided, uses the global
+            timezone variable set by the TimeUtil. A list of valid timezones can be found at
+            https://en.wikipedia.org/wiki/List_of_tz_database_time_zones. May also accept a UTC offset in seconds.
+        :return: The epoch timestamp that would display as the same time in to_timezone as the given time in
+            from_timezone.
+        """
+        millis: int = TimeUtil.format_to_millis(time_point, time_format, from_timezone)
+        return TimeUtil.shift_millis(millis, to_timezone, from_timezone)
+
     # FR-46154: Create a function that determines if a string matches a time format.
     @staticmethod
     def str_matches_format(time_point: str, time_format: str) -> bool:

sapiopycommons/multimodal/multimodal.py

@@ -0,0 +1,146 @@
+# Multimodal registration client
+from __future__ import annotations
+
+import io
+from weakref import WeakValueDictionary
+
+from databind.json import dumps, loads
+from sapiopylib.rest.User import SapioUser
+
+from sapiopycommons.general.exceptions import SapioException
+from sapiopycommons.multimodal.multimodal_data import *
+
+
+class MultiModalManager:
+    _user: SapioUser
+
+    __instances: WeakValueDictionary[SapioUser, MultiModalManager] = WeakValueDictionary()
+    __initialized: bool
+
+    def __new__(cls, user: SapioUser):
+        """
+        Observes singleton pattern per record model manager object.
+
+        :param user: The user that will make the webservice request to the application.
+        """
+        obj = cls.__instances.get(user)
+        if not obj:
+            obj = object.__new__(cls)
+            obj.__initialized = False
+            cls.__instances[user] = obj
+        return obj
+
+    def __init__(self, user:SapioUser):
+        if self.__initialized:
+            return
+        self._user = user
+        self.__initialized = True
+
+    def load_image_data(self, request: ImageDataRequestPojo) -> list[str]:
+        """
+        Loading of image data of a compound or a reaction in Sapio's unified drawing format.
+        :param request:
+        :return:
+        """
+        payload = dumps(request, ImageDataRequestPojo)
+        response = self._user.plugin_post("chemistry/request_image_data",
+                                          payload=payload, is_payload_plain_text=True)
+        self._user.raise_for_status(response)
+        return response.json()
+
+    def load_compounds(self, request: CompoundLoadRequestPojo):
+        """
+        Load compounds from the provided data here.
+        The compounds will not be registered but returned to you "the script".
+        To complete registration, you need to call register_compounds method after obtaining result.
+        """
+        payload = dumps(request, CompoundLoadRequestPojo)
+        response = self._user.plugin_post("chemistry/load",
+                                          payload=payload, is_payload_plain_text=True)
+        self._user.raise_for_status(response)
+        return loads(response.text, PyMoleculeLoaderResult)
+
+    def register_compounds(self, request: ChemRegisterRequestPojo) -> ChemCompleteImportPojo:
+        """
+        Register the filled compounds that are previously loaded via load_compounds operation.
+        """
+        payload = dumps(request, ChemRegisterRequestPojo)
+        response = self._user.plugin_post("chemistry/register",
+                                          payload=payload, is_payload_plain_text=True)
+        self._user.raise_for_status(response)
+        return loads(response.text, ChemCompleteImportPojo)
+
+    def load_reactions(self, reaction_str: str) -> PyIndigoReactionPojo:
+        """
+        Load a reaction and return the loaded reaction result.
+        :param reaction_str: A reaction string, in format of mrv, rxn, or smiles.
+        """
+        response = self._user.plugin_post("chemistry/reaction/load",
+                                          payload=reaction_str, is_payload_plain_text=True)
+        self._user.raise_for_status(response)
+        return loads(response.text, PyIndigoReactionPojo)
+
+    def register_reactions(self, reaction_str: str) -> DataRecord:
+        """
+        Register a single reaction provided.
+        Note: if the rxn has already specified a 2D coordinate, it may not be recomputed when generating record image.
+        :param reaction_str: The rxn of a reaction.
+        :return: The registered data record. This can be a record that already exists or new.
+        """
+        response = self._user.plugin_post("chemistry/reaction/register",
+                                          payload=reaction_str, is_payload_plain_text=True)
+        self._user.raise_for_status(response)
+        return loads(response.text, DataRecord)
+
+    def search_structures(self, request: ChemSearchRequestPojo) -> ChemSearchResponsePojo:
+        """
+        Perform structure search against the Sapio registries.
+        An error can be thrown as exception if search is structurally invalid.
+        :param request: The request object containing the detailed context of this search.
+        :return: The response object of the result.
+        """
+        payload = dumps(request, ChemSearchRequestPojo)
+        response = self._user.plugin_post("chemistry/search",
+                                          payload=payload, is_payload_plain_text=True)
+        self._user.raise_for_status(response)
+        return loads(response.text, ChemSearchResponsePojo)
+
+    def run_multi_sequence_alignment(self, request: MultiSequenceAlignmentRequestPojo) -> list[MultiSequenceAlignmentSeqPojo]:
+        """
+        Run a multi-sequence alignment using the specified tool and strategy.
+        :param request: The request object containing the sequences and alignment parameters. The parameters inside it can be the pojo dict of one of the options.
+        :return: The result of the multi-sequence alignment.
+        """
+        payload = dumps(request, MultiSequenceAlignmentRequestPojo)
+        response = self._user.plugin_post("bio/multisequencealignment",
+                                          payload=payload, is_payload_plain_text=True)
+        self._user.raise_for_status(response)
+        return loads(response.text, list[MultiSequenceAlignmentSeqPojo])
+
+    def register_bio(self, request: BioFileRegistrationRequest) -> BioFileRegistrationResponse:
+        """
+        Register to bioregistry of a file.
+        """
+        payload = dumps(request, BioFileRegistrationRequest)
+        response = self._user.plugin_post("bio/register/file", payload=payload, is_payload_plain_text=True)
+        self._user.raise_for_status(response)
+        return loads(response.text, BioFileRegistrationResponse)
+
+    def export_to_sdf(self, request: ChemExportSDFRequest) -> str:
+        """
+        Export the SDF files
+        :param request: The request for exporting SDF file.
+        :return: the SDF plain text data.
+        """
+        payload = dumps(request, ChemExportSDFRequest)
+        response = self._user.plugin_post("chemistry/export_sdf", payload=payload, is_payload_plain_text=True)
+        self._user.raise_for_status(response)
+        gzip_base64: str = response.text
+        if not gzip_base64:
+            raise SapioException("Returning data from server is blank for export SDF.")
+        decoded_bytes = base64.b64decode(gzip_base64)
+        with io.BytesIO(decoded_bytes) as bytes_io:
+            import gzip
+            with gzip.GzipFile(fileobj=bytes_io, mode='rb') as f:
+                ret: str = f.read().decode()
+                return ret