check_remote_data.py 840 B

1234567891011121314151617181920212223242526272829
  1. import os
  2. import paramiko
  3. import dotenv
  4. repo_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
  5. dotenv.load_dotenv(dotenv_path=os.path.join(repo_root, ".env"))
  6. host = os.environ.get('SERVER_HOST')
  7. user = os.environ.get('SERVER_USER')
  8. password = os.environ.get('SERVER_PASS')
  9. if password == "your_db_password_here" or not password:
  10. password = None
  11. ssh = paramiko.SSHClient()
  12. ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
  13. try:
  14. ssh.connect(host, username=user, password=password, timeout=10)
  15. print("SSH Connected!")
  16. cmd = "wc -l food_project/data/*.csv"
  17. stdin, stdout, stderr = ssh.exec_command(cmd)
  18. print("STDOUT:\n", stdout.read().decode('utf-8'))
  19. print("STDERR:\n", stderr.read().decode('utf-8'))
  20. except Exception as e:
  21. print(f"Error: {e}")
  22. finally:
  23. ssh.close()