"""
  离线任务统一调度脚本
  按顺序运行所有离线任务，生成推荐系统所需的各种索引
  """
  import os
  import sys
  import subprocess
  import argparse
  import logging
  from datetime import datetime
  
  # 添加父目录到路径以导入配置
  parent_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
  sys.path.insert(0, parent_dir)
  
  from offline_tasks.config.offline_config import (
      DEFAULT_LOOKBACK_DAYS,
      DEFAULT_I2I_TOP_N,
      DEFAULT_INTEREST_TOP_N
  )
  
  # 设置日志
  LOG_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'logs')
  os.makedirs(LOG_DIR, exist_ok=True)
  
  logging.basicConfig(
      level=logging.INFO,
      format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
      handlers=[
          logging.FileHandler(os.path.join(LOG_DIR, f'run_all_{datetime.now().strftime("%Y%m%d")}.log')),
          logging.StreamHandler()
      ]
  )
  logger = logging.getLogger(__name__)
  
  # 脚本目录
  SCRIPTS_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'scripts')
  
  
  def run_script(script_name, args=None):
      """
      运行Python脚本
      
      Args:
          script_name: 脚本名称
          args: 命令行参数列表
      
      Returns:
          bool: 是否成功
      """
      script_path = os.path.join(SCRIPTS_DIR, script_name)
      
      if not os.path.exists(script_path):
          logger.error(f"Script not found: {script_path}")
          return False
      
      cmd = [sys.executable, script_path]
      if args:
          cmd.extend(args)
      
      logger.info(f"Running: {' '.join(cmd)}")
      
      try:
          result = subprocess.run(
              cmd,
              check=True,
              capture_output=True,
              text=True
          )
          logger.info(f"Script {script_name} completed successfully")
          logger.debug(result.stdout)
          return True
      except subprocess.CalledProcessError as e:
          logger.error(f"Script {script_name} failed with return code {e.returncode}")
          logger.error(f"Error output: {e.stderr}")
          return False
      except Exception as e:
          logger.error(f"Unexpected error running {script_name}: {e}")
          return False
  
  
  def main():
      parser = argparse.ArgumentParser(description='Run all offline recommendation tasks')
      parser.add_argument('--skip-i2i', action='store_true', help='Skip i2i tasks')
      parser.add_argument('--skip-interest', action='store_true', help='Skip interest aggregation')
      parser.add_argument('--only-swing', action='store_true', help='Run only Swing algorithm')
      parser.add_argument('--only-w2v', action='store_true', help='Run only Session W2V')
      parser.add_argument('--only-deepwalk', action='store_true', help='Run only DeepWalk')
      parser.add_argument('--only-content', action='store_true', help='Run only Content-based similarity')
      parser.add_argument('--only-interest', action='store_true', help='Run only interest aggregation')

      parser.add_argument('--lookback_days', type=int, default=DEFAULT_LOOKBACK_DAYS, 

                          help=f'Lookback days (default: {DEFAULT_LOOKBACK_DAYS}, adjust in offline_config.py)')

      parser.add_argument('--top_n', type=int, default=DEFAULT_I2I_TOP_N, 

                          help=f'Top N similar items (default: {DEFAULT_I2I_TOP_N})')

      parser.add_argument('--debug', action='store_true',
                          help='Enable debug mode for all tasks (detailed logs + readable output files)')

      
      args = parser.parse_args()
      
      logger.info("="*80)
      logger.info("Starting offline recommendation tasks")

      if args.debug:
          logger.info("🐛 DEBUG MODE ENABLED - 详细日志 + 明文输出")

      logger.info("="*80)
      
      success_count = 0
      total_count = 0
      
      # i2i 行为相似任务
      if not args.skip_i2i:
          # 1. Swing算法
          if not args.only_w2v and not args.only_deepwalk and not args.only_interest and not args.only_content:
              logger.info("\n" + "="*80)
              logger.info("Task 1: Running Swing algorithm for i2i similarity")
              logger.info("="*80)
              total_count += 1

              script_args = [

                  '--lookback_days', str(args.lookback_days),
                  '--top_n', str(args.top_n),
                  '--time_decay'

              ]
              if args.debug:
                  script_args.append('--debug')
              if run_script('i2i_swing.py', script_args):

                  success_count += 1
          
          # 2. Session W2V
          if not args.only_swing and not args.only_deepwalk and not args.only_interest and not args.only_content:
              logger.info("\n" + "="*80)
              logger.info("Task 2: Running Session Word2Vec for i2i similarity")
              logger.info("="*80)
              total_count += 1

              script_args = [

                  '--lookback_days', str(args.lookback_days),
                  '--top_n', str(args.top_n),
                  '--save_model'

              ]
              if args.debug:
                  script_args.append('--debug')
              if run_script('i2i_session_w2v.py', script_args):

                  success_count += 1
          
          # 3. DeepWalk
          if not args.only_swing and not args.only_w2v and not args.only_interest and not args.only_content:
              logger.info("\n" + "="*80)
              logger.info("Task 3: Running DeepWalk for i2i similarity")
              logger.info("="*80)
              total_count += 1

              script_args = [

                  '--lookback_days', str(args.lookback_days),
                  '--top_n', str(args.top_n),
                  '--save_model',
                  '--save_graph'

              ]
              if args.debug:
                  script_args.append('--debug')
              if run_script('i2i_deepwalk.py', script_args):

                  success_count += 1
          
          # 4. Content-based similarity
          if not args.only_swing and not args.only_w2v and not args.only_deepwalk and not args.only_interest:
              logger.info("\n" + "="*80)
              logger.info("Task 4: Running Content-based similarity")
              logger.info("="*80)
              total_count += 1

              script_args = [

                  '--top_n', str(args.top_n),
                  '--method', 'hybrid'

              ]
              if args.debug:
                  script_args.append('--debug')
              if run_script('i2i_content_similar.py', script_args):

                  success_count += 1
      
      # 兴趣点聚合任务
      if not args.skip_interest:
          if not args.only_swing and not args.only_w2v and not args.only_deepwalk and not args.only_content:
              logger.info("\n" + "="*80)
              logger.info("Task 5: Running interest aggregation")
              logger.info("="*80)
              total_count += 1

              script_args = [

                  '--lookback_days', str(args.lookback_days),
                  '--top_n', str(DEFAULT_INTEREST_TOP_N)

              ]
              if args.debug:
                  script_args.append('--debug')
              if run_script('interest_aggregation.py', script_args):

                  success_count += 1
      
      # 总结
      logger.info("\n" + "="*80)
      logger.info(f"All tasks completed: {success_count}/{total_count} succeeded")
      logger.info("="*80)
      
      if success_count == total_count:
          logger.info("✓ All tasks completed successfully!")
          return 0
      else:
          logger.warning(f"✗ {total_count - success_count} task(s) failed")
          return 1
  
  
  if __name__ == '__main__':
      sys.exit(main())