Merge pull request #145 from mdsol/feature/add_location_oid_type

Feature/add location oid type

Author: glow-mdsol

.github/workflows/main.yml

@@ -11,13 +11,13 @@ on:
 jobs:
   tests:
     name: "Python ${{ matrix.python-version }}"
-    runs-on: "ubuntu-latest"
+    runs-on: ubuntu-latest
     env:
-      USING_COVERAGE: '3.7,3.8,3.10'
+      USING_COVERAGE: '3.7,3.8,3.10,3.11,3.12'
 
     strategy:
       matrix:
-        python-version: ["3.7", "3.8", "3.9", "3.10", "3.11"]
+        python-version: ["3.7", "3.8", "3.9", "3.10", "3.11", "3.12"]
 
     steps:
       - uses: actions/checkout@v3
@@ -60,7 +60,7 @@ jobs:
       # Install dependencies. `--no-root` means "install all dependencies but not the project
       # itself", which is what you want to avoid caching _your_ code. The `if` statement
       # ensures this only runs on a cache miss.
-      - run: poetry install --no-interaction --no-root
+      - run: poetry install --no-interaction --no-root --with dev
         if: steps.cache-deps.outputs.cache-hit != 'true'
 
       # Now install _your_ project. This isn't necessary for many types of projects -- particularly

CONTRIBUTING.md

@@ -9,14 +9,14 @@ Developer Setup
    ```
    git clone https://github.com/mdsol/rwslib
    ```
-2. Create a Virtual Env for the Local instance
+2. Install poetry and start a shell
     ```bash
-    $ python -m venv venv
-    $ source venv/bin/activate
+    $ pip install poetry
+    $ poetry shell
     ```
 3. Install the development dependencies
     ```bash
-    $ pip install -r requirements-dev.txt
+    $ poetry install --with dev
     ```
 4. Enjoy !!!
 

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rwslib"
-version = "1.2.11"
+version = "1.2.12"
 description = "Rave Web Services for Python"
 authors = ["Ian Sparks <[email protected]>"]
 maintainers = ["Geoff Low <[email protected]>"]
@@ -14,11 +14,9 @@ classifiers = [
     'Natural Language :: English',
     'License :: OSI Approved :: MIT License',
     'Programming Language :: Python',
-    'Programming Language :: Python :: 3.7',
-    'Programming Language :: Python :: 3.8',
-    'Programming Language :: Python :: 3.9',
     'Programming Language :: Python :: 3.10',
     'Programming Language :: Python :: 3.11',
+    'Programming Language :: Python :: 3.12',
 ]
 packages = [
     { include = "rwslib" },
@@ -33,19 +31,21 @@ click = "*"
 faker = "*"
 urllib3 = "*"
 
+[tool.poetry.group.dev]
+optional = true
 
 [tool.poetry.group.dev.dependencies]
 httpretty = "*"
 mock = "*"
 coverage = "*"
 pytest = "*"
 tox = "*"
-tox-poetry = "*"
 
 [tool.poetry.group.docs.dependencies]
 Sphinx = "^5.1.1"
 sphinx-pyproject = "^0.3.0"
 
 [build-system]
-requires = ["poetry-core"]
+requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"
+

pyproject.toml

@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+from typing import Optional
+
 from rwslib.builders.common import (
     ODMElement,
     TransactionalElement,
@@ -8,7 +10,7 @@
 )
 from rwslib.builders.modm import LastUpdateMixin, MilestoneMixin
 from rwslib.builders.metadata import MeasurementUnitRef
-from rwslib.builders.constants import ProtocolDeviationStatus, QueryStatusType
+from rwslib.builders.constants import ProtocolDeviationStatus, QueryStatusType, LocationOIDType
 from rwslib.builders.common import Unset
 
 from collections import OrderedDict
@@ -77,6 +79,7 @@ def __init__(
         subject_key,
         subject_key_type="SubjectName",
         transaction_type="Update",
+        location_oid_type: Optional[LocationOIDType] = None,
     ):
         """
         :param str site_location_oid: :class:`SiteLocation` OID
@@ -98,6 +101,8 @@ def __init__(
         self.signature = None
         #: :class:`SiteRef`
         self.siteref = None
+        #: :class:`LocationOIDType`
+        self.location_oid_type = location_oid_type
 
     def build(self, builder):
         """Build XML by appending to builder"""
@@ -120,8 +125,8 @@ def build(self, builder):
         if self.siteref:
             self.siteref.build(builder)
         else:
-            builder.start("SiteRef", {"LocationOID": self.sitelocationoid})
-            builder.end("SiteRef")
+            _siteref = SiteRef(self.sitelocationoid, self.location_oid_type)
+            _siteref.build(builder)
 
         for event in self.study_events:
             event.build(builder)
@@ -993,17 +998,66 @@ class SiteRef(ODMElement, LastUpdateMixin):
     .. note:: The `mdsol:LocationOIDType` attribute should be used to indicate the type of `LocationOID`
     """
 
-    def __init__(self, oid):
+    def __init__(self,
+                 oid,
+                 location_oid_type: Optional[LocationOIDType] = None,
+                 site_number: Optional[str] = None,
+                 study_site_name: Optional[str] = None,
+                 site_uuid: Optional[str] = None,
+                 location_name: Optional[str] = None,
+                 study_env_site_number: Optional[str] = None,
+                 previous_location_oid: Optional[str] = None,
+                 shared_location_oid: Optional[str] = None,
+                 previous_shared_location_oid: Optional[str] = None,
+                 subject_created_at_site: Optional[str] = None,):
         """
         :param str oid: OID for referenced :class:`Location`
+        :param LocationOIDType location_oid_type: Type for the oid
+        :param site_number: Site number
+        :param study_site_name: Study site name
+        :param site_uuid: Site UUID
+        :param location_name: Location name
+        :param study_env_site_number: Study environment site number
+        :param previous_location_oid: Previous location OID
+        :param shared_location_oid: Shared location OID
+        :param previous_shared_location_oid: Previous shared location OID
+        :param subject_created_at_site: Subject created at site
         """
         self.oid = oid
+        self.location_oid_type = location_oid_type
+        self.site_number = site_number
+        self.study_site_name = study_site_name
+        self.site_uuid = site_uuid
+        self.location_name = location_name
+        self.study_env_site_number = study_env_site_number
+        self.previous_location_oid = previous_location_oid
+        self.shared_location_oid = shared_location_oid
+        self.previous_shared_location_oid = previous_shared_location_oid
+        self.subject_created_at_site = subject_created_at_site
 
     def build(self, builder):
         """
         Build XML by appending to builder
         """
         params = dict(LocationOID=self.oid)
+        if self.location_oid_type:
+            params["mdsol:LocationOIDType"] = self.location_oid_type.value
+        elif self.site_uuid:
+            params["mdsol:SiteUUID"] = self.site_uuid
+        elif self.site_number:
+            params["mdsol:SiteNumber"] = self.site_number
+        elif self.study_site_name:
+            params["mdsol:StudySiteName"] = self.study_site_name
+        elif self.study_env_site_number:
+            params["mdsol:StudyEnvSiteNumber"] = self.study_env_site_number
+        if self.location_name:
+            params["mdsol:LocationName"] = self.location_name
+        if self.previous_location_oid:
+            params["mdsol:PreviousLocationOID"] = self.previous_location_oid
+        if self.shared_location_oid:
+            params["mdsol:SharedLocationOID"] = self.shared_location_oid
+        if self.subject_created_at_site:
+            params["mdsol:SubjectCreatedAtSite"] = self.subject_created_at_site
         # mixins
         self.mixin()
         self.mixin_params(params)

rwslib/builders/clinicaldata.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
-
+import platform
 from string import ascii_letters
-from datetime import datetime
+import datetime
 from xml.etree import cElementTree as ET
 
 
@@ -13,11 +13,21 @@
 
 # -----------------------------------------------------------------------------------------------------------------------
 # Utilities
+def get_utc_date() -> datetime.datetime:
+    """
+    Returns the UTC date as datetime.datetime object.
+    """
+    version = platform.python_version_tuple()
+    if int(version[1]) < 11:
+        utc_date = datetime.datetime.utcnow()
+    else:
+        utc_date = datetime.datetime.now(datetime.UTC)
+    return utc_date
 
 
 def now_to_iso8601():
     """Returns NOW date/time as a UTC date/time formated as iso8601 string"""
-    utc_date = datetime.utcnow()
+    utc_date = get_utc_date()
     return dt_to_iso8601(utc_date)
 
 

rwslib/builders/common.py

@@ -329,3 +329,14 @@ class GranularityType(enum.Enum):
     AllClinicalData = 'AllClinicalData'
     SingleSite = 'SingleSite'
     SingleSubject = 'SingleSubject'
+
+
+class LocationOIDType(enum.Enum):
+    """
+    Location OID Type Enumeration
+    Applies to a :class:`SiteRef`
+    """
+    SiteUUID = 'SiteUUID'
+    SiteNumber = 'SiteNumber'
+    StudyEnvSiteNumber = 'StudyEnvSiteNumber'
+

rwslib/builders/constants.py

@@ -5,6 +5,8 @@
 import datetime
 import enum
 
+from rwslib.builders.common import get_utc_date
+
 
 class MODMExtensionRegistry(enum.Enum):
     """
@@ -167,7 +169,7 @@ def set_update_time(self, update_time=None):
         if update_time and isinstance(update_time, (datetime.datetime,)):
             self.last_update_time = update_time
         else:
-            self.last_update_time = datetime.datetime.utcnow()
+            self.last_update_time = get_utc_date()
 
     def mixin_params(self, params):
         """