Update dags/ods/ig_post_insights/udfs.py

Co-authored-by: Wei Lee <[email protected]>
pycontw · Aug 17, 2024 · 76b4987 · 76b4987
1 parent fbb48ce
commit 76b4987
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/dags/ods/ig_post_insights/udfs.py b/dags/ods/ig_post_insights/udfs.py
@@ -56,14 +56,11 @@ def save_posts_and_insights() -> None:
 
     last_post = query_last_post()
     print(last_post)
-    if last_post is None:
-        new_posts = posts
-    else:
-        new_posts = [
-            post
-            for post in posts
-            if post["timestamp"] > last_post["created_at"].timestamp()
-        ]
+    new_posts = [
+        post
+        for post in posts
+        if post["timestamp"] > last_post["created_at"].timestamp()
+    ] if last_post else posts
 
     if not dump_posts_to_bigquery(
         [