Skip to content

Commit a05055a

Browse files
sombraSoftmigueldalberto
authored andcommitted
Format uploadMetadata.py with black
Signed-off-by: Joao Luis Sombrio <[email protected]>
1 parent 32a77c2 commit a05055a

1 file changed

Lines changed: 17 additions & 5 deletions

File tree

flow/util/uploadMetadata.py

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,13 @@
1010
import firebase_admin
1111
from firebase_admin import credentials
1212
from firebase_admin import firestore
13+
1314
# --- END FIRESTORE ---
1415

1516
# --- PUBSUB ---
1617
from google.cloud import pubsub_v1
1718
from 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

257265
RUN_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

Comments
 (0)