tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-06 19:25:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(entity): change metadata data type to text, json etc

Browse Source
This commit is contained in:
arkohut 2024-06-11 15:18:00 +08:00
parent 107f7d06c2
commit b0becf64ea
4 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
memos/fixtures/patch_entity_metadata_response.json
View File

@ -10,7 +10,7 @@
"library_id": 1, "library_id": 1,
"metadata_entries": [ "metadata_entries": [
{ {
"data_type": "attribute", "data_type": "text",
"entity_id": 1, "entity_id": 1,
"id": 1, "id": 1,
"key": "author", "key": "author",
@ -18,7 +18,7 @@
"value": "Jane Smith" "value": "Jane Smith"
}, },
{ {
"data_type": "attribute", "data_type": "text",
"entity_id": 1, "entity_id": 1,
"id": 2, "id": 2,
"key": "year", "key": "year",

4
memos/schemas.py
View File

@ -11,8 +11,8 @@ class MetadataSource(Enum):
class MetadataType(Enum): class MetadataType(Enum):
EXTRACONTENT = "extra_content" JSON_DATA = "json"
ATTRIBUTE = "attribute" TEXT_DATA = "text"
class NewLibraryParam(BaseModel): class NewLibraryParam(BaseModel):

4
memos/server.py
View File

@ -98,7 +98,7 @@ def new_folder(
return crud.add_folder(library_id=library.id, folder=folder, db=db) return crud.add_folder(library_id=library.id, folder=folder, db=db)
async def trigger_webhooks(library, entity, request): async def trigger_webhooks(library: Library, entity: Entity, request: Request):
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
tasks = [] tasks = []
for plugin in library.plugins: for plugin in library.plugins:
@ -110,7 +110,7 @@ async def trigger_webhooks(library, entity, request):
) )
task = client.post( task = client.post(
plugin.webhook_url, plugin.webhook_url,
json={"entity": entity.model_dump(mode="json")}, json=entity.model_dump(mode="json"),
headers={"Location": location}, headers={"Location": location},
timeout=10.0, # Adding a timeout of 10 seconds timeout=10.0, # Adding a timeout of 10 seconds
) )

10
memos/test_server.py
View File

@ -474,7 +474,7 @@ def test_add_metadata_entry_to_entity_success(client):
key="author", key="author",
value="John Doe", value="John Doe",
source="plugin_generated", source="plugin_generated",
data_type=MetadataType.ATTRIBUTE, data_type=MetadataType.TEXT_DATA,
) )
update_entity_param = UpdateEntityParam(attrs=[metadata_entry]) update_entity_param = UpdateEntityParam(attrs=[metadata_entry])
@ -530,13 +530,13 @@ def test_patch_entity_metadata_entries(client):
"key": "author", "key": "author",
"value": "Jane Smith", "value": "Jane Smith",
"source": "user_generated", "source": "user_generated",
"data_type": MetadataType.ATTRIBUTE.value, "data_type": MetadataType.TEXT_DATA.value,
}, },
{ {
"key": "year", "key": "year",
"value": "2023", "value": "2023",
"source": "user_generated", "source": "user_generated",
"data_type": MetadataType.ATTRIBUTE.value, "data_type": MetadataType.TEXT_DATA.value,
}, },
] ]
update_entity_param = UpdateEntityParam( update_entity_param = UpdateEntityParam(
@ -563,7 +563,7 @@ def test_patch_entity_metadata_entries(client):
"key": "author", "key": "author",
"value": "John Doe", "value": "John Doe",
"source": "user_generated", "source": "user_generated",
"data_type": MetadataType.ATTRIBUTE.value, "data_type": MetadataType.TEXT_DATA.value,
} }
] ]
update_entity_param = UpdateEntityMetadataParam( update_entity_param = UpdateEntityMetadataParam(
@ -594,7 +594,7 @@ def test_patch_entity_metadata_entries(client):
"key": "media_type", "key": "media_type",
"value": "book", "value": "book",
"source": "user_generated", "source": "user_generated",
"data_type": MetadataType.ATTRIBUTE.value, "data_type": MetadataType.TEXT_DATA.value,
} }
updated_metadata_entries.append(new_metadata_entry) updated_metadata_entries.append(new_metadata_entry)