From ec718f0e4c522b6212202c5c34a9326dea97b206 Mon Sep 17 00:00:00 2001
From: Andrey <andrey@simiotics.com>
Date: Thu, 8 Sep 2022 17:51:01 +0300
Subject: [PATCH] Move table name from parameters to format.

---
 .../mooncrawl/mooncrawl/metadata_crawler/cli.py |  2 +-
 .../mooncrawl/mooncrawl/metadata_crawler/db.py  | 17 +++++++++++++----
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/crawlers/mooncrawl/mooncrawl/metadata_crawler/cli.py b/crawlers/mooncrawl/mooncrawl/metadata_crawler/cli.py
index 21ea4a9f..451ac0c9 100644
--- a/crawlers/mooncrawl/mooncrawl/metadata_crawler/cli.py
+++ b/crawlers/mooncrawl/mooncrawl/metadata_crawler/cli.py
@@ -126,7 +126,7 @@ def handle_crawl(args: argparse.Namespace) -> None:
 
     blockchain_type = AvailableBlockchainType(args.blockchain)
 
-    parse_metadata(blockchain_type, args.batch_size)
+    parse_metadata(blockchain_type, args.commit_batch_size)
 
 
 def main() -> None:
diff --git a/crawlers/mooncrawl/mooncrawl/metadata_crawler/db.py b/crawlers/mooncrawl/mooncrawl/metadata_crawler/db.py
index d06dbcd0..14ae5a03 100644
--- a/crawlers/mooncrawl/mooncrawl/metadata_crawler/db.py
+++ b/crawlers/mooncrawl/mooncrawl/metadata_crawler/db.py
@@ -80,14 +80,16 @@ def get_uris_of_tokens(
             address as address
 
         FROM
-            :table
+            {}
         WHERE
             label = :label
             AND label_data ->> 'name' = :name
         ORDER BY
             label_data -> 'inputs'-> 0 ASC,
             block_number :: INT DESC;
-    """,
+    """.format(
+            table
+        ),
         {"table": table, "label": VIEW_STATE_CRAWLER_LABEL, "name": "tokenURI"},
     )
 
@@ -111,18 +113,25 @@ def get_current_metadata_for_address(
     """
     Get existing metadata.
     """
+
+    label_model = get_label_model(blockchain_type)
+
+    table = label_model.__tablename__
+
     current_metadata = db_session.execute(
         """ SELECT
             DISTINCT ON(label_data ->> 'token_id') label_data ->> 'token_id' as token_id
         FROM
-            polygon_labels
+            {}
         WHERE
             address = :address
             AND label = :label
         ORDER BY
             label_data ->> 'token_id' ASC,
             block_number :: INT DESC;
-    """,
+    """.format(
+            table
+        ),
         {"address": address, "label": METADATA_CRAWLER_LABEL},
     )