Fix NPE bug in torrentinfo set call

Fixes 747494304
This commit is contained in:
Joseph Schorr 2018-10-30 15:39:57 -04:00
parent 436e8cb760
commit 366bddd20a
2 changed files with 12 additions and 4 deletions

View file

@ -338,10 +338,14 @@ def get_storage_locations(uuid):
def save_torrent_info(storage_object, piece_length, pieces):
try:
return TorrentInfo.create(storage=storage_object, piece_length=piece_length, pieces=pieces)
except IntegrityError:
# TorrentInfo already exists for this storage.
pass
return TorrentInfo.get(storage=storage_object, piece_length=piece_length)
except TorrentInfo.DoesNotExist:
try:
return TorrentInfo.create(storage=storage_object, piece_length=piece_length, pieces=pieces)
except IntegrityError:
# TorrentInfo already exists for this storage.
return TorrentInfo.get(storage=storage_object, piece_length=piece_length)
def get_torrent_info(blob):
try: