def DrinkToProto(drink, full=False):
ret = models_pb2.Drink()
ret.id = drink.id
ret.url = drink.get_absolute_url()
ret.ticks = drink.ticks
ret.volume_ml = drink.volume_ml
ret.session_id = drink.session_id
ret.time = datestr(drink.time)
ret.duration = drink.duration
if drink.keg:
ret.keg_id = drink.keg_id
if drink.user_id:
ret.user_id = drink.user.username
if drink.shout:
ret.shout = drink.shout
if drink.tick_time_series:
ret.tick_time_series = drink.tick_time_series
if full:
if drink.user_id:
ret.user.MergeFrom(ToProto(drink.user))
if drink.keg:
ret.keg.MergeFrom(ToProto(drink.keg))
if drink.session:
ret.session.MergeFrom(ToProto(drink.session))
if drink.picture:
ret.images.add().MergeFrom(ToProto(drink.picture))
return ret
def DrinkToProto(drink, full=False):
ret = models_pb2.Drink()
ret.id = drink.seqn
ret.url = drink.get_absolute_url()
ret.ticks = drink.ticks
ret.volume_ml = drink.volume_ml
ret.session_id = drink.session.seqn
ret.time = datestr(drink.time)
ret.duration = drink.duration
ret.status = drink.status
if drink.keg:
ret.keg_id = drink.keg.seqn
if drink.user:
ret.user_id = drink.user.username
if drink.auth_token:
ret.auth_token_id = str(drink.auth_token.id)
if drink.shout:
ret.shout = drink.shout
if full:
if drink.user:
ret.user.MergeFrom(ToProto(drink.user))
if drink.keg:
ret.keg.MergeFrom(ToProto(drink.keg))
if drink.session:
ret.session.MergeFrom(ToProto(drink.session))
for i in drink.pictures.all():
ret.images.add().MergeFrom(ToProto(i))
return ret
