check_remote_ps.py 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. import os
  2. import sys
  3. import paramiko
  4. from dotenv import load_dotenv
  5. sys.stdout.reconfigure(encoding='utf-8')
  6. sys.stderr.reconfigure(encoding='utf-8')
  7. repo_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
  8. load_dotenv(dotenv_path=os.path.join(repo_root, ".env"))
  9. host = os.environ.get('SERVER_HOST')
  10. user = os.environ.get('SERVER_USER')
  11. password = os.environ.get('SERVER_PASS')
  12. if password == "your_db_password_here" or not password:
  13. password = None
  14. ssh = paramiko.SSHClient()
  15. ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  16. try:
  17. ssh.connect(host, username=user, password=password, timeout=10)
  18. print("SSH Connected!")
  19. print("--- Running Containers ---")
  20. stdin, stdout, stderr = ssh.exec_command("docker ps --format 'table {{.Names}}\t{{.Status}}\t{{.Command}}'")
  21. print(stdout.read().decode('utf-8'))
  22. print("--- Streamlit App Logs ---")
  23. db_cmd = "docker logs --tail 100 food_project-app-1"
  24. stdin, stdout, stderr = ssh.exec_command(db_cmd)
  25. print(stdout.read().decode('utf-8'))
  26. print(stderr.read().decode('utf-8'))
  27. print("--- Ingest Container Logs ---")
  28. stdin, stdout, stderr = ssh.exec_command("docker ps -a --filter name=food_project-ingest-run -q | xargs -I {} docker logs {}")
  29. print(stdout.read().decode('utf-8'))
  30. print(stderr.read().decode('utf-8'))
  31. except Exception as e:
  32. print(f"Error: {e}")
  33. finally:
  34. ssh.close()