Track bundled vendor runtime sources

vendor/social-auto-upload/utils/files_times.py

@@ -0,0 +1,83 @@
+from datetime import timedelta
+
+from datetime import datetime
+from pathlib import Path
+
+from conf import BASE_DIR
+
+
+def get_absolute_path(relative_path: str, base_dir: str = None) -> str:
+    # Convert the relative path to an absolute path
+    absolute_path = Path(BASE_DIR) / base_dir / relative_path
+    return str(absolute_path)
+
+
+def get_title_and_hashtags(filename):
+    """
+  获取视频标题和 hashtag
+
+  Args:
+    filename: 视频文件名
+
+  Returns:
+    视频标题和 hashtag 列表
+  """
+
+    # 获取视频标题和 hashtag txt 文件名
+    txt_filename = filename.replace(".mp4", ".txt")
+
+    # 读取 txt 文件
+    with open(txt_filename, "r", encoding="utf-8") as f:
+        content = f.read()
+
+    # 获取标题和 hashtag
+    splite_str = content.strip().split("\n")
+    title = splite_str[0]
+    hashtags = splite_str[1].replace("#", "").split(" ")
+
+    return title, hashtags
+
+
+def generate_schedule_time_next_day(total_videos, videos_per_day = 1, daily_times=None, timestamps=False, start_days=0):
+    """
+    Generate a schedule for video uploads, starting from the next day.
+
+    Args:
+    - total_videos: Total number of videos to be uploaded.
+    - videos_per_day: Number of videos to be uploaded each day.
+    - daily_times: Optional list of specific times of the day to publish the videos.
+    - timestamps: Boolean to decide whether to return timestamps or datetime objects.
+    - start_days: Start from after start_days.
+
+    Returns:
+    - A list of scheduling times for the videos, either as timestamps or datetime objects.
+    """
+    if videos_per_day <= 0:
+        raise ValueError("videos_per_day should be a positive integer")
+
+    if daily_times is None:
+        # Default times to publish videos if not provided
+        daily_times = [6, 11, 14, 16, 22]
+
+    if videos_per_day > len(daily_times):
+        raise ValueError("videos_per_day should not exceed the length of daily_times")
+
+    # Generate timestamps
+    schedule = []
+    current_time = datetime.now()
+
+    for video in range(total_videos):
+        day = video // videos_per_day + start_days + 1  # +1 to start from the next day
+        daily_video_index = video % videos_per_day
+
+        # Calculate the time for the current video
+        hour = daily_times[daily_video_index]
+        time_offset = timedelta(days=day, hours=hour - current_time.hour, minutes=-current_time.minute,
+                                seconds=-current_time.second, microseconds=-current_time.microsecond)
+        timestamp = current_time + time_offset
+
+        schedule.append(timestamp)
+
+    if timestamps:
+        schedule = [int(time.timestamp()) for time in schedule]
+    return schedule