142 lines
5.9 KiB
Python
142 lines
5.9 KiB
Python
import os
|
||
import datetime
|
||
from django.core.management.base import BaseCommand
|
||
from django.utils import timezone
|
||
from user_management.models import PlatformAccount, Video
|
||
from django.conf import settings
|
||
|
||
class Command(BaseCommand):
|
||
help = '测试视频上传和定时发布功能'
|
||
|
||
def add_arguments(self, parser):
|
||
parser.add_argument('video_path', type=str, help='视频文件路径')
|
||
parser.add_argument('platform_account_id', type=int, help='平台账号ID')
|
||
parser.add_argument('--title', type=str, help='视频标题(可选)')
|
||
parser.add_argument('--desc', type=str, help='视频描述(可选)')
|
||
parser.add_argument('--schedule', type=str, help='计划发布时间，格式: YYYY-MM-DD HH:MM:SS (可选)')
|
||
|
||
def handle(self, *args, **options):
|
||
video_path = options['video_path']
|
||
platform_account_id = options['platform_account_id']
|
||
title = options.get('title')
|
||
desc = options.get('desc')
|
||
schedule_str = options.get('schedule')
|
||
|
||
# 验证视频文件是否存在
|
||
if not os.path.exists(video_path):
|
||
self.stderr.write(self.style.ERROR(f'错误: 视频文件不存在: {video_path}'))
|
||
return
|
||
|
||
# 验证平台账号是否存在
|
||
try:
|
||
platform_account = PlatformAccount.objects.get(id=platform_account_id)
|
||
except PlatformAccount.DoesNotExist:
|
||
self.stderr.write(self.style.ERROR(f'错误: 未找到ID为{platform_account_id}的平台账号'))
|
||
return
|
||
|
||
# 设置标题（如果未提供，则使用文件名）
|
||
if not title:
|
||
title = os.path.splitext(os.path.basename(video_path))[0]
|
||
|
||
# 准备保存视频的目录
|
||
media_root = getattr(settings, 'MEDIA_ROOT', os.path.join(settings.BASE_DIR, 'media'))
|
||
videos_dir = os.path.join(media_root, 'videos')
|
||
account_dir = os.path.join(videos_dir, f"{platform_account.platform_name}_{platform_account.account_name}")
|
||
|
||
if not os.path.exists(videos_dir):
|
||
os.makedirs(videos_dir)
|
||
if not os.path.exists(account_dir):
|
||
os.makedirs(account_dir)
|
||
|
||
# 生成唯一的文件名
|
||
import time
|
||
timestamp = int(time.time())
|
||
file_name = f"{timestamp}_{os.path.basename(video_path)}"
|
||
file_path = os.path.join(account_dir, file_name)
|
||
|
||
# 复制视频文件
|
||
with open(video_path, 'rb') as src_file:
|
||
with open(file_path, 'wb') as dest_file:
|
||
dest_file.write(src_file.read())
|
||
|
||
self.stdout.write(self.style.SUCCESS(f'视频文件已复制到: {file_path}'))
|
||
|
||
# 创建视频记录
|
||
video_data = {
|
||
'platform_account': platform_account,
|
||
'title': title,
|
||
'description': desc,
|
||
'local_path': file_path,
|
||
'status': 'draft',
|
||
}
|
||
|
||
# 处理计划发布时间
|
||
if schedule_str:
|
||
try:
|
||
from dateutil import parser
|
||
scheduled_time = parser.parse(schedule_str)
|
||
|
||
# 如果时间已过，设置为当前时间后5分钟
|
||
now = timezone.now()
|
||
if scheduled_time <= now:
|
||
scheduled_time = now + datetime.timedelta(minutes=5)
|
||
self.stdout.write(self.style.WARNING(
|
||
f'警告: 计划时间已过，已调整为当前时间后5分钟: {scheduled_time}'
|
||
))
|
||
|
||
video_data['scheduled_time'] = scheduled_time
|
||
video_data['status'] = 'scheduled'
|
||
|
||
except Exception as e:
|
||
self.stderr.write(self.style.ERROR(f'错误: 解析时间失败: {str(e)}'))
|
||
return
|
||
|
||
# 创建视频对象
|
||
video = Video.objects.create(**video_data)
|
||
|
||
self.stdout.write(self.style.SUCCESS(f'创建视频记录成功，ID: {video.id}'))
|
||
|
||
# 如果是计划发布，创建定时任务
|
||
if video.status == 'scheduled':
|
||
try:
|
||
from django_celery_beat.models import PeriodicTask, CrontabSchedule
|
||
import json
|
||
|
||
scheduled_time = video.scheduled_time
|
||
|
||
# 创建定时任务
|
||
schedule, _ = CrontabSchedule.objects.get_or_create(
|
||
minute=scheduled_time.minute,
|
||
hour=scheduled_time.hour,
|
||
day_of_month=scheduled_time.day,
|
||
month_of_year=scheduled_time.month,
|
||
)
|
||
|
||
# 创建周期性任务
|
||
task_name = f"Publish_Video_{video.id}_{time.time()}"
|
||
PeriodicTask.objects.create(
|
||
name=task_name,
|
||
task='user_management.tasks.publish_scheduled_video',
|
||
crontab=schedule,
|
||
args=json.dumps([video.id]),
|
||
one_off=True, # 只执行一次
|
||
start_time=scheduled_time
|
||
)
|
||
|
||
self.stdout.write(self.style.SUCCESS(
|
||
f'创建定时发布任务成功，计划发布时间: {scheduled_time}'
|
||
))
|
||
|
||
except Exception as e:
|
||
self.stderr.write(self.style.ERROR(f'创建定时任务失败: {str(e)}'))
|
||
|
||
self.stdout.write(self.style.SUCCESS('操作完成'))
|
||
|
||
# 打印执行方式提示
|
||
if video.status == 'scheduled':
|
||
self.stdout.write(f"\n视频将在 {video.scheduled_time} 自动发布")
|
||
self.stdout.write("\n要手动发布，可以使用以下命令:")
|
||
else:
|
||
self.stdout.write("\n要发布该视频，可以使用以下命令:")
|
||
|
||
self.stdout.write(f"python manage.py publish_video {video.id}") |