Added JournalTypes to be able to specify humbug and default

Author: kompotkot

bugout/app.py

@@ -348,10 +348,16 @@ def delete_journal_scopes(
 
     # Journal handlers
     def create_journal(
-        self, token: Union[str, uuid.UUID], name: str, timeout: float = REQUESTS_TIMEOUT
+        self,
+        token: Union[str, uuid.UUID],
+        name: str,
+        journal_type: Union[str, data.JournalTypes],
+        timeout: float = REQUESTS_TIMEOUT,
     ) -> data.BugoutJournal:
         self.journal.timeout = timeout
-        return self.journal.create_journal(token=token, name=name)
+        return self.journal.create_journal(
+            token=token, name=name, journal_type=data.JournalTypes(journal_type)
+        )
 
     def list_journals(
         self, token: Union[str, uuid.UUID], timeout: float = REQUESTS_TIMEOUT

bugout/data.py

@@ -33,6 +33,11 @@ class HolderType(Enum):
     group = "group"
 
 
+class JournalTypes(Enum):
+    DEFAULT = "default"
+    HUMBUG = "humbug"
+
+
 class BugoutUser(BaseModel):
     id: uuid.UUID = Field(alias="user_id")
     username: str

bugout/journal.py

@@ -14,6 +14,7 @@
     BugoutSearchResults,
     HolderType,
     Method,
+    JournalTypes,
 )
 from .settings import REQUESTS_TIMEOUT
 
@@ -107,11 +108,14 @@ def delete_journal_scopes(
         return BugoutJournalScopeSpecs(**result)
 
     # Journal module
-    def create_journal(self, token: Union[str, uuid.UUID], name: str) -> BugoutJournal:
+    def create_journal(
+        self,
+        token: Union[str, uuid.UUID],
+        name: str,
+        journal_type: JournalTypes = JournalTypes.DEFAULT,
+    ) -> BugoutJournal:
         journal_path = "journals/"
-        json = {
-            "name": name,
-        }
+        json = {"name": name, "journal_type": journal_type.value}
         headers = {
             "Authorization": f"Bearer {token}",
         }