Skip to content

Commit

Permalink
add video updates
Browse files Browse the repository at this point in the history
  • Loading branch information
DGaffney committed Dec 18, 2023
1 parent 7eb1b0e commit 7c6fb6b
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 5 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -122,8 +122,8 @@ Output Message:
"id": "A unique string ID that identifies the item being processed",
"callback_url": "A unique URL that will be requested upon completion",
"url": "The URL at which the media is located",
"bucket": "bucket within which the .tmk file is stored",
"outfile": "The filename of the .tmk file generated for the video",
"folder": "bucket within which the .tmk file is stored",
"filepath": "The filename of the .tmk file generated for the video",
"hash_value": "The shorter, getPureAverageFeature hash from tmk (used in first-pass approximation searches)",
}
```
Expand Down
2 changes: 1 addition & 1 deletion lib/model/video.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,4 +58,4 @@ def process(self, video: schemas.Message) -> schemas.GenericItem:
s3.upload_file_to_s3(self.tmk_bucket(), self.tmk_file_path(video_filename))
finally:
os.remove(temp_file_name)
return dict(**video.dict(), **{"bucket": self.tmk_bucket(), "outfile": self.tmk_file_path(video_filename), "hash_value": hash_value})
return dict(**video.dict(), **{"folder": self.tmk_bucket(), "filepath": self.tmk_file_path(video_filename), "hash_value": hash_value})
4 changes: 2 additions & 2 deletions lib/schemas.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@ class MediaItem(GenericItem):
hash_value: Optional[Any] = None

class VideoItem(MediaItem):
bucket: Optional[str] = None
outfile: Optional[str] = None
folder: Optional[str] = None
filepath: Optional[str] = None

class Message(BaseModel):
body: Union[MediaItem, VideoItem]
Expand Down

0 comments on commit 7c6fb6b

Please sign in to comment.