87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
import json, csv, os, time, stomp, logging
|
|
from datetime import datetime
|
|
from flask_backend.Stomp_client import get_connection
|
|
|
|
QUEUE_NAME = "/queue/testQueue"
|
|
OUTPUT_DIR = "/app/output"
|
|
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
|
|
|
# 🔧 Logger setup
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(levelname)s:%(name)s:%(message)s"
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class MyListener(stomp.ConnectionListener):
|
|
def on_message(self, frame):
|
|
logger.info(f"Přijatá zpráva z activeMQ: {frame.body}")
|
|
|
|
try:
|
|
data = json.loads(frame.body)
|
|
except Exception as e:
|
|
logger.error(f"Error when creating JSON: {e}\nInput was: {frame}")
|
|
return
|
|
|
|
# Vytvoření souboru
|
|
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
|
filename = os.path.join(OUTPUT_DIR, f"actors_{timestamp}.csv")
|
|
|
|
"""try:
|
|
with open(filename, "w", newline="", encoding="utf-8") as f:
|
|
writer = csv.writer(f, delimiter=";")
|
|
|
|
for actor in data:
|
|
row = ["list"]
|
|
|
|
for key, value in actor.items():
|
|
value = actor.get(key, "")
|
|
|
|
if isinstance(value, list):
|
|
for item in value:
|
|
if isinstance(item, dict):
|
|
row.extend(item.values())
|
|
else:
|
|
row.append(str(item))
|
|
|
|
|
|
elif isinstance(value, bool):
|
|
value = str(value).lower()
|
|
|
|
else:
|
|
row.append(value)
|
|
|
|
writer.writerow(row)
|
|
|
|
logger.info(f"CSV uložen: {filename} ({len(data)} řádků)")
|
|
except Exception as e:
|
|
logger.error(f"Chyba při zápisu CSV: {e}")"""
|
|
|
|
try:
|
|
with open(filename, "w", newline="", encoding="utf-8") as f:
|
|
writer = csv.writer(f, delimiter=";")
|
|
|
|
for actor in data:
|
|
row = ["list"]
|
|
for key, value in actor.items():
|
|
if isinstance(value, dict): # rozbalíme dict (např. address)
|
|
row.extend(value.values())
|
|
else:
|
|
row.append(value)
|
|
writer.writerow(row)
|
|
|
|
logger.info(f"CSV uložen: {filename} ({len(data)} řádků)")
|
|
except Exception as e:
|
|
logger.error(f"Chyba při zápisu CSV: {e}")
|
|
|
|
# ActiveMQ client
|
|
listener = MyListener()
|
|
conn = get_connection(listener=listener)
|
|
conn.subscribe(destination=QUEUE_NAME, id=1, ack="auto")
|
|
|
|
logger.info(f"ActiveMQ is running\nAddress: {QUEUE_NAME}")
|
|
|
|
# Keep the process active
|
|
while True:
|
|
time.sleep(1)
|