def test_saving_new_channel(self): channel = Channel( title="Channel Title", url="https://www.youtube.com/channel/UCsn8UgBuRxGGqKmrAy5d3gA", ) channel.save() saved_channels = Channel.objects.all() self.assertEqual(saved_channels.count(), 1)
def resetChannelList(hdid, channelList): device = Device.objects.get(hdid=hdid) Channel.objects.filter(device=device).delete() current_channel = '' channel_pattern = re.compile('SCANNING.*us-cable:(\d+)') program_pattern = re.compile('PROGRAM\s(\d+):\s(?![0])(.*)') for line in channelList: m = channel_pattern.match(line) if m: current_channel = m.group(1) else: m = program_pattern.match(line) if m: c = Channel(device=device, channel=int(current_channel), program=int(m.group(1)), desc=m.group(2)) c.save()
def import_file(channel_name, file): ''' Import csv file to db. ''' with open(file) as f: reader = csv.reader(f) try: channel = Channel.objects.get(name=channel_name) except Channel.DoesNotExist: channel = Channel(name=channel_name) channel.save() for row in reader: if row[0] != 'Category': categories = row[0].split(' / ') i = 0 category_objs = [] while i < len(categories): try: if i == 0: channel_category = ChannelCategory.objects.get( name=categories[i], channel=channel, parent_category=None) else: channel_category = ChannelCategory.objects.get( name=categories[i], channel=channel, parent_category=category_objs[i - 1]) except ChannelCategory.DoesNotExist: channel_category = ChannelCategory() channel_category.name = categories[i] channel_category.channel = channel if i != 0: channel_category.parent_category = category_objs[i - 1] channel_category.save() category_objs.append(channel_category) i += 1
def writeToDataBase(self): ch = Channel(name=self.title, country=self.country, subscriberCount=self.subscriberCount) ch.save()