def handle(self, *args, **options):
if len(args) > 2:
raise CommandError('Too many arguments given. Provide either one or two usernames.')
elif len(args) == 2:
try:
free_user = User(username=args[0], first_name=args[0] + '_free', last_name='User', email='*****@*****.**')
paid_user = User(username=args[1], first_name=args[1] + '_paid', last_name='User', email='*****@*****.**')
free_user.save()
paid_user.save()
self.stdout.write('Successfully created fake users using supplied usernames.\n')
except:
raise CommandError("Could not create users with supplied usernames. Try different usernames.")
elif len(args) == 1:
try:
free_user = User(username=args[0] + '_free', first_name=args[0] + '_free', last_name='User', email='*****@*****.**')
paid_user = User(username=args[0] + '_paid', first_name=args[0] + '_paid', last_name='User', email='*****@*****.**')
free_user.save()
paid_user.save()
self.stdout.write('Successfully created fake users using the supplied username + suffixes "_free" and "_paid".\n')
except:
raise CommandError("Could not create users with supplied username. Try a different username.")
elif len(args) == 0:
try:
free_user = User(username='******', first_name='fakeuser_free', last_name='User', email='*****@*****.**')
paid_user = User(username='******', first_name='fakeuser_paid', last_name='User', email='*****@*****.**')
free_user.save()
paid_user.save()
self.stdout.write('Successfully created fake users using default usernames.\n')
except:
raise CommandError("Fake users already exist. Create differently named users using 'manage.py create_test_accounts <username>'")
free_user.set_password('password')
paid_user.set_password('password')
free_user.save()
paid_user.save()
# Set paid_user account
paid_level = AccountLevel.objects.get(level='Paid')
paid_user.userprofile.account_level = paid_level
paid_user.userprofile.save()
# Create videos for free users
for u in (free_user, paid_user):
created_videos = ()
for i in range(10):
temp_slug = generate_slug(5)
temp_vid = Video(uploader=u, title="Test Fake Uploads Video " + str(i), slug=temp_slug)
temp_vid.save()
created_videos = created_videos + (temp_vid,)
#### Videos
#Normal
temp_vid = created_videos[0]
temp_vid.title = "Normal private video"
temp_vid.description = "Normal private video."
temp_vid.save()
upload_to_s3(temp_vid.slug)
### Expired videos, for free user
temp_vid = created_videos[1]
if not u == free_user:
upload_to_s3(temp_vid.slug)
temp_vid = created_videos[2]
upload_to_s3(temp_vid.slug)
temp_vid = created_videos[3]
upload_to_s3(temp_vid.slug)
#Expired Yesterday, created 2 days ago
temp_vid = created_videos[1]
temp_vid.title = "Expired the day previous to creation"
temp_vid.description = "Expired the day previous to creation."
temp_vid.created = datetime.datetime.now() - datetime.timedelta(days=2)
temp_vid.save()
if u == free_user:
temp_vid.expired = True
temp_vid.expiry_date = datetime.datetime.now() - datetime.timedelta(days=1)
temp_vid.save()
temp_vid.delete()
#Expires Today in 4 hours
temp_vid = created_videos[2]
temp_vid.title = "Expires 4 hours from creation"
temp_vid.description = "Expires 4 hours from creation."
if u == free_user:
temp_vid.expiry_date = datetime.datetime.now() + datetime.timedelta(hours=4)
temp_vid.save()
#Expires in 3 days
temp_vid = created_videos[3]
temp_vid.title = "Expires 3 days from creation"
temp_vid.description = "Expires 3 days from creation."
if u == free_user:
temp_vid.expiry_date = datetime.datetime.now() + datetime.timedelta(days=3)
temp_vid.save()
#Created Yesterday
temp_vid = created_videos[4]
upload_to_s3(temp_vid.slug)
temp_vid.created = datetime.datetime.now() - datetime.timedelta(days=1)
temp_vid.title = "Time-stamped as created the day previous to actual creation"
temp_vid.description = "Time-stamped as created the day previous to actual creation."
temp_vid.save()
#Created A Week Previous
temp_vid = created_videos[5]
upload_to_s3(temp_vid.slug)
temp_vid.created = datetime.datetime.now() - datetime.timedelta(days=7)
temp_vid.title = "Time-stamped as created a week previous to actual creation"
temp_vid.description = "Time-stamped as created a week previous to actual creation."
temp_vid.save()
#Public
temp_vid = created_videos[6]
upload_to_s3(temp_vid.slug)
temp_vid.is_public = True
temp_vid.title = "Public."
temp_vid.description = "Public."
temp_vid.save()
#Public. Expired Yesterday, created 2 days ago
temp_vid = created_videos[7]
upload_to_s3(temp_vid.slug)
temp_vid.title = "Public. Expired the day previous to creation"
temp_vid.description = "Public. Expired the day previous to creation."
temp_vid.is_public = True
temp_vid.created = datetime.datetime.now() - datetime.timedelta(days=2)
if u == free_user:
temp_vid.expired = True
temp_vid.expiry_date = datetime.datetime.now() - datetime.timedelta(days=1)
temp_vid.save()
#Outlined
temp_vid = created_videos[8]
upload_to_s3(temp_vid.slug)
temp_vid.title = "Outlined"
temp_vid.description = "Outlined."
temp_vid.save()
outline = VideoOutline(video=temp_vid)
outline.save()
outline_pin = VideoOutlinePin(video_outline=outline, text="Start")
outline_pin.save()
#Transferred to Youtube, created 2 days ago, deleted from server yesterday
temp_vid = created_videos[9]
upload_to_s3(temp_vid.slug)
temp_vid.youtube_embed_url= "http://www.youtube.com/embed/yc5W-AClSlI"
temp_vid.created = datetime.datetime.now() - datetime.timedelta(days=2)
temp_vid.youtube_video_expiry_date = datetime.datetime.now() - datetime.timedelta(days=1)
temp_vid.title = "YouTube video"
temp_vid.description = "YouTube video."
temp_vid.save()
self.stdout.write('Successfully created videos!\n')
