| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364 |
- import os
- import sys
- import paramiko
- import dotenv
- import time
- sys.stdout.reconfigure(encoding='utf-8')
- sys.stderr.reconfigure(encoding='utf-8')
- repo_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
- dotenv.load_dotenv(dotenv_path=os.path.join(repo_root, ".env"))
- host = os.environ.get('SERVER_HOST')
- user = os.environ.get('SERVER_USER')
- password = os.environ.get('SERVER_PASS')
- if password == "your_db_password_here" or not password:
- password = None
- ssh = paramiko.SSHClient()
- ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
- try:
- ssh.connect(host, username=user, password=password, timeout=10)
- print("SSH Connected!")
-
- # 1. Stop app container to release all locks
- print("Stopping app container...")
- stdin, stdout, stderr = ssh.exec_command("cd food_project && docker-compose stop app")
- stdout.read()
-
- # 2. Restart MySQL to drop any blocked index creation queries
- print("Restarting MySQL...")
- stdin, stdout, stderr = ssh.exec_command("cd food_project && docker-compose restart mysql")
- stdout.read()
-
- print("Waiting 10 seconds for MySQL to be healthy...")
- time.sleep(10)
-
- # 3. Create full-text indexes
- queries = [
- "ALTER TABLE food_db.products_core ADD FULLTEXT INDEX idx_prod_ing (product_name, ingredients_text);",
- "ALTER TABLE food_db.products_core ADD FULLTEXT INDEX idx_prod_name (product_name);",
- "ALTER TABLE food_db.products_core ADD FULLTEXT INDEX idx_ing_text (ingredients_text);"
- ]
-
- db_pass = os.environ.get('MYSQL_ROOT_PASSWORD') or 'your_mysql_root_pass'
- for q in queries:
- print(f"Executing: {q}")
- cmd = f"docker exec food_project-mysql-1 mysql -uroot -p{db_pass} -e '{q}'"
- stdin, stdout, stderr = ssh.exec_command(cmd)
- print("STDOUT:\n", stdout.read().decode('utf-8'))
- print("STDERR:\n", stderr.read().decode('utf-8'))
-
- # 4. Start app container back up
- print("Starting app container...")
- stdin, stdout, stderr = ssh.exec_command("cd food_project && docker-compose start app")
- stdout.read()
-
- print("Index creation completed successfully!")
-
- except Exception as e:
- print(f"Error: {e}")
- finally:
- ssh.close()
|