def setUp(self): super(SniptResourceTest, self).setUp() # Johnny self.johnny = User.objects.create_user('johnny', '*****@*****.**', 'password') ApiKey.objects.get_or_create(user=self.johnny) self.johnny_auth = self.create_apikey(self.johnny, self.johnny.api_key.key) self.johnny_private = Snipt(title='Private snipt for Johnny', lexer='text', public=False, user=self.johnny) self.johnny_public = Snipt(title='Public snipt for Johnny', lexer='text', public=True, user=self.johnny) self.johnny_private.save() self.johnny_public.save() # Bob self.bob = User.objects.create_user('bob', '*****@*****.**', 'password') ApiKey.objects.get_or_create(user=self.bob) self.bob_auth = self.create_apikey(self.bob, self.bob.api_key.key) self.bob_private = Snipt(title='Private snipt for Bob', lexer='text', public=False, user=self.bob) self.bob_public = Snipt(title='Public snipt for Bob', lexer='text', public=True, user=self.bob) self.bob_private.save() self.bob_public.save()
def handle(self, *args, **options): api_key = options["api_key"][0] from_username = options["from_username"][0] to_username = options["to_username"][0] to_user = User.objects.get(username=to_username) print(u"Fetching snipts...") items = get_snipts(api_key, from_username) items.reverse() for snipt in items: s = Snipt( blog_post=snipt["blog_post"], code=snipt["code"], description=snipt["description"], id=snipt["id"], key=snipt["key"], lexer=snipt["lexer"], line_count=snipt["line_count"], meta=snipt["meta"], public=snipt["public"], publish_date=snipt["publish_datetime"], secure=snipt["secure"], slug=snipt["slug"], stylized=snipt["stylized"], title=snipt["title"], user=to_user, views=snipt["views"], ) s.created = snipt["created"] s.modified = snipt["modified"] for tag in snipt["tags"]: s.tags.add(tag["name"]) s.save() self.stdout.write(snipt["title"])