kopia lustrzana https://github.com/bugout-dev/moonstream
Change report routes.
rodzic
5e5869d99d
commit
3c61d78c83
|
@ -476,21 +476,23 @@ def generate_report_nft_dashboard_handler(args: argparse.Namespace):
|
||||||
|
|
||||||
client = Moonstream()
|
client = Moonstream()
|
||||||
|
|
||||||
|
reports = {
|
||||||
|
"cu_nft_dashboard_data": {},
|
||||||
|
"cu_land_nft_dashboard_data": {},
|
||||||
|
"cu_seaport_feed_polygon": {},
|
||||||
|
"cu_breeding_feed": {},
|
||||||
|
"cu_shadowcorns_owners": {},
|
||||||
|
"cu_shadowcorns_feed": {},
|
||||||
|
"cu_previous_day_distributed_rewards": {},
|
||||||
|
"cu_nft_dashboard_data_xai": {"blockchain": "xai"},
|
||||||
|
}
|
||||||
|
|
||||||
for query in client.list_queries(
|
for query in client.list_queries(
|
||||||
token=args.moonstream_token,
|
token=args.moonstream_token,
|
||||||
).queries:
|
).queries:
|
||||||
params = {} # type: ignore
|
params = {} # type: ignore
|
||||||
|
|
||||||
if query.name not in [
|
if query.name not in reports:
|
||||||
"cu_nft_dashboard_data",
|
|
||||||
"cu_land_nft_dashboard_data",
|
|
||||||
"cu_seaport_feed_polygon",
|
|
||||||
"cu_breeding_feed",
|
|
||||||
"cu_shadowcorns_owners",
|
|
||||||
"cu_shadowcorns_feed",
|
|
||||||
"cu_previous_day_distributed_rewards",
|
|
||||||
"cu_nft_dashboard_data_xai",
|
|
||||||
]:
|
|
||||||
continue
|
continue
|
||||||
try:
|
try:
|
||||||
logger.info(f"Generating report for {query.name}")
|
logger.info(f"Generating report for {query.name}")
|
||||||
|
@ -507,9 +509,17 @@ def generate_report_nft_dashboard_handler(args: argparse.Namespace):
|
||||||
# send as json
|
# send as json
|
||||||
ext = "json"
|
ext = "json"
|
||||||
|
|
||||||
|
blockchain = ""
|
||||||
|
|
||||||
|
if reports[query.name].get("blockchain"):
|
||||||
|
|
||||||
|
blockchain = "/" + reports[query.name]["blockchain"]
|
||||||
|
|
||||||
|
path = f"queries/CryptoUnicorns{blockchain}/{query.name}/data.{ext}"
|
||||||
|
|
||||||
url = client.upload_query_results(
|
url = client.upload_query_results(
|
||||||
json.dumps(data),
|
json.dumps(data),
|
||||||
key=f"queries/{query.name}/data.{ext}",
|
key=path,
|
||||||
bucket=MOONSTREAM_S3_PUBLIC_DATA_BUCKET,
|
bucket=MOONSTREAM_S3_PUBLIC_DATA_BUCKET,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -518,6 +528,8 @@ def generate_report_nft_dashboard_handler(args: argparse.Namespace):
|
||||||
logger.info(f"Data recived. Uploading report for {query.name} as csv")
|
logger.info(f"Data recived. Uploading report for {query.name} as csv")
|
||||||
|
|
||||||
ext = "csv"
|
ext = "csv"
|
||||||
|
|
||||||
|
path = f"queries/CU{blockchain}/{query.name}/data.{ext}"
|
||||||
csv_buffer = StringIO()
|
csv_buffer = StringIO()
|
||||||
|
|
||||||
dict_csv_writer = csv.DictWriter(
|
dict_csv_writer = csv.DictWriter(
|
||||||
|
@ -530,7 +542,7 @@ def generate_report_nft_dashboard_handler(args: argparse.Namespace):
|
||||||
|
|
||||||
url = client.upload_query_results(
|
url = client.upload_query_results(
|
||||||
data=csv_buffer.getvalue().encode("utf-8"),
|
data=csv_buffer.getvalue().encode("utf-8"),
|
||||||
key=f"queries/{query.name}/data.{ext}",
|
key=path,
|
||||||
bucket=MOONSTREAM_S3_PUBLIC_DATA_BUCKET,
|
bucket=MOONSTREAM_S3_PUBLIC_DATA_BUCKET,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
Ładowanie…
Reference in New Issue