fix tooling

Author: mariajgrimaldi

openedx_events/learning/data.py

@@ -0,0 +1,77 @@
+"""
+Data attributes for events within the architecture subdomain `learning`. These
+attributes follow the form of attr objects specified in OEP-49 data pattern.
+"""
+
+from datetime import datetime
+from typing import Dict
+
+import attr
+from opaque_keys.edx.keys import CourseKey
+
+
+@attr.s(frozen=True)
+class ProfileData:
+    """
+    Attributes defined for Open edX student's profile object.
+    """
+    meta = attr.ib(type=Dict[str, str], factory=dict)
+    name = attr.ib(type=str, factory=str)
+
+
+@attr.s(frozen=True)
+class StudentData:
+    """
+    Attributes defined for Open edX student object.
+    """
+    username = attr.ib(type=str)
+    email = attr.ib(type=str)
+    first_name = attr.ib(type=str, factory=str)
+    last_name = attr.ib(type=str, factory=str)
+    is_active = attr.ib(type=bool, default=False)
+    profile = attr.ib(type=ProfileData, default=ProfileData())
+
+
+@attr.s(frozen=True)
+class RegistrationData:
+    """
+    Attributes defined for Open edX registration profile object.
+    """
+    activation_key = attr.ib(type=str)
+
+
+@attr.s(frozen=True)
+class CourseData:
+    """
+    Attributes defined for Open edX course object.
+    """
+    course_key = attr.ib(type=CourseKey)
+    display_name = attr.ib(type=CourseKey, factory=str)
+
+
+@attr.s(frozen=True)
+class EnrollmentData:
+    """
+    Attributes defined for Open edX Course Enrollment object.
+    """
+    user = attr.ib(type=StudentData)
+    course = attr.ib(type=CourseData)
+    mode = attr.ib(type=str)
+
+
+@attr.s(frozen=True)
+class CertificateData(EnrollmentData):
+    """
+    Attributes defined for Open edX Course Enrollment object.
+    """
+    status = attr.ib(type=str)
+    available_date = attr.ib(type=datetime)
+
+
+@attr.s(frozen=True)
+class CohortData:
+    """
+    Attributes defined for Open edX Course Enrollment object.
+    """
+    user = attr.ib(type=StudentData)
+    course = attr.ib(type=CourseData)

openedx_events/tooling.py

@@ -16,7 +16,7 @@ class OpenEdxPublicSignal(Signal):
     """Custom class used to create Open edX events.
     """
 
-    def __init__(self, event_type=None, data=None, minor_version='0.0'):
+    def __init__(self, event_type=None, data=None, minor_version="0.0"):
         if not event_type or not data:
             raise InstantiationException(event_type=event_type)
         self.data = data
@@ -36,7 +36,7 @@ def get_signal_metadata(self):
         def get_current_time():
             """Helper function used to get timestamp when the event ocurred.
             """
-            return str(datetime.utcnow().isoformat())  + 'Z'
+            return str(datetime.utcnow().isoformat())  + "Z"
 
         def get_source():
             """Helper function used to get logical source of an event.