from util.neo4j_utils import neo4j_client


def get_drug_count():
    cypher = "MATCH (d:Drug) WHERE d.name IS NOT NULL RETURN count(d) AS total"
    res = neo4j_client.execute_read(cypher)
    return res[0]["total"]

# 在路由中临时加一行打印
print("Total drug names:", get_drug_count())  # 看是否接近 17000+