Merge pull request #15 from BlockScience/types

Google Drive File, FileAuthor, & Export RIDs

Author: lukvmil

src/rid_lib/types/__init__.py

@@ -10,4 +10,7 @@
 from .discord_guild import DiscordGuild
 from .discord_channel import DiscordChannel
 from .discord_message import DiscordMessage
-from .discord_user import DiscordUser
+from .discord_user import DiscordUser
+from .gdrive_file import GoogleDriveFile
+from .gdrive_file_authors import GoogleDriveFileAuthors
+from .gdrive_export import GoogleDriveExport

src/rid_lib/types/gdrive_export.py

@@ -0,0 +1,31 @@
+from rid_lib.core import ORN
+
+
+class GoogleDriveExport(ORN):
+    namespace = 'google_drive.export'
+    
+    def __init__(self, type: str, subtype: str, fileId: str):
+        self.type = type
+        self.subtype = subtype
+        self.fileId = fileId
+
+    @property
+    def mimeType(self):
+        return f'{self.type}/{self.subtype}'
+    
+    @property
+    def reference(self):
+        return f'{self.mimeType}/{self.fileId}'
+    
+    @classmethod
+    def from_reference(cls, reference):
+        if type(reference) is str:
+            components = reference.split("/")
+            if len(components) == 3:
+                return cls(*components)
+            else:
+                raise ValueError(
+                    "Google Drive File Export reference must contain 3 '/'-separated components: '<type>/<subtype>/<fileId>'"
+                )
+        else:
+            raise ValueError("Google Drive File Export reference must be a string")

src/rid_lib/types/gdrive_file.py

@@ -0,0 +1,36 @@
+from rid_lib.core import ORN
+from rid_lib.types.gdrive_export import GoogleDriveExport
+
+
+class GoogleDriveFile(ORN):
+    namespace = 'google_drive.file'
+    mimeType = 'application/vnd.google-apps.file'
+
+    def __init__(self, id: str):
+        self.id = id
+
+    @property
+    def url(self):
+        return f'https://drive.google.com/file/d/{self.id}'
+    
+    @property
+    def export(self, type: str, subtype: str) -> GoogleDriveExport:
+        return GoogleDriveExport(
+            type=type, 
+            subtype=subtype, 
+            fileId=self.id
+        )
+
+    @property
+    def reference(self):
+        return self.id
+
+    @classmethod
+    def from_reference(cls, id: str):
+        if type(id) is str:
+            if len(id) >= 1:
+                return cls(id)
+            else:
+                raise ValueError("Google Drive File reference MUST NOT be an empty string")
+        else:
+            raise ValueError("Google Drive File reference MUST be a string")

src/rid_lib/types/gdrive_file_authors.py

@@ -0,0 +1,27 @@
+from rid_lib.core import ORN
+from rid_lib.types.gdrive_file import GoogleDriveFile
+
+
+class GoogleDriveFileAuthors(ORN):
+    namespace = 'google_drive.file.authors'
+
+    def __init__(self, fileId: str):
+        self.fileId = fileId
+
+    @property
+    def file(self) -> GoogleDriveFile:
+        return GoogleDriveFile(self.fileId)
+
+    @property
+    def reference(self):
+        return self.fileId
+            
+    @classmethod
+    def from_reference(cls, fileId: str):
+        if type(fileId) is str:
+            if len(fileId) >= 1:
+                return cls(fileId)
+            else:
+                raise ValueError("Google Drive File Authors reference must not be an empty string")
+        else:
+            raise ValueError("Google Drive File Authors reference must be a string")