vbatts/quay
Archived
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

implement get_torrent_info

Browse source
This commit is contained in:
Jimmy Zelinskie 2015-12-31 14:09:50 -05:00
parent a9b7ac6b48
commit 9b0a84c02f
2 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
data/model/__init__.py
View file

@ -8,6 +8,9 @@ class DataModelException(Exception):
class BlobDoesNotExist(DataModelException): class BlobDoesNotExist(DataModelException):
pass pass
class TorrentInfoDoesNotExist(DataModelException):
pass
class InvalidBlobUpload(DataModelException): class InvalidBlobUpload(DataModelException):
pass pass

12
data/model/storage.py
View file

@ -2,7 +2,7 @@ import logging
from peewee import JOIN_LEFT_OUTER, fn, SQL from peewee import JOIN_LEFT_OUTER, fn, SQL
from data.model import config, db_transaction, InvalidImageException from data.model import config, db_transaction, InvalidImageException, TorrentInfoDoesNotExist
from data.database import (ImageStorage, Image, DerivedStorageForImage, ImageStoragePlacement, from data.database import (ImageStorage, Image, DerivedStorageForImage, ImageStoragePlacement,
ImageStorageLocation, ImageStorageTransformation, ImageStorageSignature, ImageStorageLocation, ImageStorageTransformation, ImageStorageSignature,
ImageStorageSignatureKind, Repository, Namespace, TorrentInfo) ImageStorageSignatureKind, Repository, Namespace, TorrentInfo)
@ -216,3 +216,13 @@ def get_storage_locations(uuid):
def save_torrent_info(storage_object, piece_length, pieces): def save_torrent_info(storage_object, piece_length, pieces):
TorrentInfo.create(storage=storage_object, piece_length=piece_length, pieces=pieces) TorrentInfo.create(storage=storage_object, piece_length=piece_length, pieces=pieces)
def get_torrent_info(blob):
try:
return (TorrentInfo
.select()
.join(ImageStorage)
.where(blob.id == ImageStorage.id)
.get())
except TorrentInfo.DoesNotExist:
raise TorrentInfoDoesNotExist