1010import firebase_admin
1111from firebase_admin import credentials
1212from firebase_admin import firestore
13+
1314# --- END FIRESTORE ---
1415
1516# --- PUBSUB ---
1617from google .cloud import pubsub_v1
1718from google .oauth2 import service_account
19+
1820# --- END PUBSUB ---
1921
2022# make sure the working dir is flow/
@@ -192,7 +194,9 @@ def upload_data(db, dataFile, platform, design, variant, args, rules):
192194
193195
194196# --- PUBSUB ---
195- def publish_to_pubsub (publisher , topic_path , dataFile , platform , design , variant , args , rules ):
197+ def publish_to_pubsub (
198+ publisher , topic_path , dataFile , platform , design , variant , args , rules
199+ ):
196200 """Publish a single design's metrics to Pub/Sub as a JSON message."""
197201 with open (dataFile ) as f :
198202 data = json .load (f )
@@ -215,7 +219,9 @@ def publish_to_pubsub(publisher, topic_path, dataFile, platform, design, variant
215219 message_data = json .dumps (payload ).encode ("utf-8" )
216220 future = publisher .publish (topic_path , data = message_data )
217221 message_id = future .result ()
218- print (f"[INFO] Published to Pub/Sub (message ID: { message_id } ) for { platform } { design } { variant } ." )
222+ print (
223+ f"[INFO] Published to Pub/Sub (message ID: { message_id } ) for { platform } { design } { variant } ."
224+ )
219225
220226
221227# --- END PUBSUB ---
@@ -251,7 +257,9 @@ def get_rules(dataFile):
251257 # No credentials file — use default credentials (e.g., emulator or ADC)
252258 publisher = pubsub_v1 .PublisherClient ()
253259 topic_path = publisher .topic_path (args .pubsubProjectID , args .pubsubTopicID )
254- print (f"[INFO] Pub/Sub publisher initialized (default creds) for topic: { topic_path } " )
260+ print (
261+ f"[INFO] Pub/Sub publisher initialized (default creds) for topic: { topic_path } "
262+ )
255263# --- END PUBSUB init ---
256264
257265RUN_FILENAME = "metadata.json"
@@ -273,7 +281,9 @@ def get_rules(dataFile):
273281 print (f"[WARN] Skiping upload { platform } { design } { variant } ." )
274282 continue
275283 print (f"[INFO] Get rules for { platform } { design } { variant } ." )
276- rules = get_rules (os .path .join ("designs" , platform , design , f"rules-{ variant } .json" ))
284+ rules = get_rules (
285+ os .path .join ("designs" , platform , design , f"rules-{ variant } .json" )
286+ )
277287
278288 # --- FIRESTORE (remove when deprecating) ---
279289 if db :
@@ -288,5 +298,7 @@ def get_rules(dataFile):
288298 publisher , topic_path , dataFile , platform , design , variant , args , rules
289299 )
290300 except Exception as e :
291- print (f"[WARN] Pub/Sub publish failed for { platform } { design } { variant } : { e } " )
301+ print (
302+ f"[WARN] Pub/Sub publish failed for { platform } { design } { variant } : { e } "
303+ )
292304 # --- END PUBSUB ---
0 commit comments