def create(request):
device_form = DeviceForm(request.POST)
if not device_form.is_valid():
messages.error(request, _('Please fill out all fields.'))
return HttpResponseRedirect(reverse('devices'))
device = Device()
device.name = device_form.cleaned_data['name']
device.type = device_form.cleaned_data['type']
device.uid = device_form.cleaned_data['uid'].replace(' ', '-')
try:
set_device(request.user, device)
messages.success(request, _('Device saved'))
except DeviceUIDException as e:
messages.error(request, _(unicode(e)))
return HttpResponseRedirect(reverse('devices'))
except Unauthorized:
messages.error(request, _("You can't use the same Device "
"ID for two devices."))
return HttpResponseRedirect(reverse('devices'))
return HttpResponseRedirect(reverse('device-edit', args=[device.uid]))
def get_device(user, uid, user_agent, undelete=True):
"""
Loads or creates the device indicated by user, uid.
If the device has been deleted and undelete=True, it is undeleted.
"""
store_ua = user.get_wksetting(STORE_UA)
save = False
try:
device = user.get_device_by_uid(uid, only_active=False)
except DeviceDoesNotExist:
device = Device(uid=uid)
user.devices.append(device)
save = True
if device.deleted and undelete:
device.deleted = False
user.set_device(device)
save = True
if store_ua and user_agent and \
getattr(device, 'user_agent', None) != user_agent:
device.user_agent = user_agent
user.set_device(device)
save = True
if save:
user.save()
return device
def _get(user, uid, undelete):
save = False
try:
device = user.get_device_by_uid(uid, only_active=False)
except DeviceDoesNotExist:
device = Device(uid=uid)
user.devices.append(device)
save = True
if device.deleted and undelete:
device.deleted = False
user.set_device(device)
save = True
if store_ua and user_agent and getattr(device, "user_agent", None) != user_agent:
device.user_agent = user_agent
user.set_device(device)
save = True
if save:
user.save()
return device
def create(request):
device_form = DeviceForm(request.POST)
if not device_form.is_valid():
messages.error(request, _('Please fill out all fields.'))
return HttpResponseRedirect(reverse('device-edit-new'))
device = Device()
device.name = device_form.cleaned_data['name']
device.type = device_form.cleaned_data['type']
device.uid = device_form.cleaned_data['uid'].replace(' ', '-')
try:
request.user.set_device(device)
request.user.save()
messages.success(request, _('Device saved'))
except DeviceUIDException as e:
messages.error(request, _(str(e)))
return render(request, 'device-create.html', {
'device': device,
'device_form': device_form,
})
except:
messages.error(request, _("You can't use the same Device "
"ID for two devices."))
return render(request, 'device-create.html', {
'device': device,
'device_form': device_form,
})
return HttpResponseRedirect(reverse('device-edit', args=[device.uid]))
def test_merge(self):
p1 = Podcast()
p1.urls = ['http://example.com/podcast1.rss']
p1.save()
p2 = Podcast()
p2.urls = ['http://example.com/podcast2.rss']
p2.save()
e1 = Episode()
e1.title = 'Episode 1'
e1.podcast = p1.get_id()
e1.urls = ['http://example.com/podcast1/e1.mp3']
e1.save()
e2 = Episode()
e2.title = 'Episode 2'
e2.podcast = p1.get_id()
e2.urls = ['http://example.com/podcast1/e2.mp3']
e2.save()
e3 = Episode()
e3.title = 'Episode 3'
e3.podcast = p2.get_id()
e3.urls = ['http://example.com/podcast2/e2.mp3']
e3.save()
e4 = Episode()
e4.title = 'Episode 4'
e4.podcast = p2.get_id()
e4.urls = ['http://example.com/podcast2/e3.mp3']
e4.save()
user = User()
user.username = '******'
user.email = '*****@*****.**'
user.set_password('secret')
device1 = Device()
device1.uid = 'dev1'
device2 = Device()
device2.uid = 'dev2'
user.devices.append(device1)
user.devices.append(device2)
user.save()
p1.subscribe(user, device1)
time.sleep(1)
p1.unsubscribe(user, device1)
time.sleep(1)
p1.subscribe(user, device1)
p2.subscribe(user, device2)
s1 = e1.get_user_state(user)
s1.add_actions([EpisodeAction(action='play')])
s1.save()
s3 = e3.get_user_state(user)
s3.add_actions([EpisodeAction(action='play')])
s3.save()
# we need that for later
e3_id = e3._id
actions = Counter()
# decide which episodes to merge
groups = [(0, [e1]), (1, [e2, e3]), (2, [e4])]
# carry out the merge
pm = PodcastMerger([p1, p2], actions, groups)
pm.merge()
e1 = Episode.get(e1._id)
es1 = e1.get_user_state(user)
self.assertEqual(len(es1.actions), 1)
# check if merged episode's id can still be accessed
e3 = Episode.get(e3_id)
es3 = e3.get_user_state(user)
self.assertEqual(len(es3.actions), 1)
p1 = Podcast.get(p1.get_id())
ps1 = p1.get_user_state(user)
self.assertEqual(len(ps1.get_subscribed_device_ids()), 2)
self.assertEqual(len(list(p1.get_episodes())), 3)
