100 lines
3.0 KiB
Python
100 lines
3.0 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
测试temperature_table.py脚本，设置正确的环境变量
|
||
"""
|
||
|
||
import os
|
||
import json
|
||
import subprocess
|
||
import sys
|
||
from datetime import datetime
|
||
|
||
def load_influx_config():
|
||
"""从default.json加载InfluxDB配置"""
|
||
config_path = os.path.join(os.path.dirname(__file__), "default.json")
|
||
try:
|
||
with open(config_path, 'r', encoding='utf-8') as f:
|
||
config = json.load(f)
|
||
return config.get('influx', {})
|
||
except Exception as e:
|
||
print(f"加载配置文件失败: {e}")
|
||
return {}
|
||
|
||
def test_temperature_table():
|
||
"""测试temperature_table.py脚本"""
|
||
print("测试 temperature_table.py 脚本")
|
||
print("=" * 50)
|
||
|
||
# 加载InfluxDB配置
|
||
influx_config = load_influx_config()
|
||
if not influx_config:
|
||
print("无法加载InfluxDB配置")
|
||
return
|
||
|
||
print("InfluxDB配置:")
|
||
print(f" URL: {influx_config.get('url', 'N/A')}")
|
||
print(f" ORG: {influx_config.get('org', 'N/A')}")
|
||
print(f" BUCKET: {influx_config.get('bucket', 'N/A')}")
|
||
print(f" MEASUREMENT: {influx_config.get('measurement', 'N/A')}")
|
||
print()
|
||
|
||
# 设置环境变量
|
||
env = os.environ.copy()
|
||
env.update({
|
||
'INFLUX_URL': influx_config.get('url', ''),
|
||
'INFLUX_ORG': influx_config.get('org', ''),
|
||
'INFLUX_TOKEN': influx_config.get('token', ''),
|
||
'INFLUX_BUCKET': influx_config.get('bucket', 'PCM'),
|
||
'INFLUX_MEASUREMENT': influx_config.get('measurement', 'PCM_Measurement'),
|
||
'EXPERIMENT_START': '2025-11-27T11:30:00Z',
|
||
'EXPERIMENT_END': '2025-11-27T12:00:00Z',
|
||
'TABLE_LOG_LEVEL': 'DEBUG',
|
||
'TABLE_LOG_FILE': 'temperature_test.log'
|
||
})
|
||
|
||
print("运行 temperature_table.py 脚本...")
|
||
print(f" 实验时间: {env['EXPERIMENT_START']} → {env['EXPERIMENT_END']}")
|
||
print()
|
||
|
||
# 运行脚本
|
||
script_path = os.path.join(os.path.dirname(__file__), "temperature_table.py")
|
||
try:
|
||
result = subprocess.run(
|
||
[sys.executable, script_path],
|
||
env=env,
|
||
capture_output=True,
|
||
text=True,
|
||
timeout=30
|
||
)
|
||
|
||
print("脚本输出:")
|
||
if result.stdout:
|
||
print("STDOUT:")
|
||
print(result.stdout)
|
||
|
||
if result.stderr:
|
||
print("STDERR:")
|
||
print(result.stderr)
|
||
|
||
print(f"返回码: {result.returncode}")
|
||
|
||
# 检查日志文件
|
||
log_file = "temperature_test.log"
|
||
if os.path.exists(log_file):
|
||
print(f"\n日志文件 ({log_file}) 最后10行:")
|
||
with open(log_file, 'r', encoding='utf-8') as f:
|
||
lines = f.readlines()
|
||
for line in lines[-10:]:
|
||
print(f" {line.strip()}")
|
||
|
||
except subprocess.TimeoutExpired:
|
||
print("脚本运行超时")
|
||
except Exception as e:
|
||
print(f"运行脚本失败: {e}")
|
||
|
||
def main():
|
||
test_temperature_table()
|
||
|
||
if __name__ == "__main__":
|
||
main()
|