def create_subscription(pigeon_id, name): pigeon_key = ndb.Key(Pigeon, pigeon_id) stream_list = Stream.query(Stream.name == name).fetch() stream_key = stream_list[0].key sub = Subscription() sub.Pigeon_key = pigeon_key sub.Stream_key = stream_key sub.put() return
def create_stream(pigeon_id, name, cover_url, sub_list, tag_list): pigeon_key = ndb.Key(Pigeon, pigeon_id) stream = Stream() stream.key = ndb.Key(Stream, name, parent=pigeon_key) stream.name = name stream.cover_url = cover_url stream.tags = tag_list stream.put() for pid in sub_list: if pid != pigeon_id: if not pigeon_exists(pid): create_pigeon(pid) suber_key = ndb.Key(Pigeon, pid) sub = Subscription() sub.Pigeon_key = suber_key sub.Stream_key = stream.key sub.put() return