Untitled

from google.cloud.spanner_v1.types import Type, TypeCode, StructType
from database import SpannerDatabase, MockSpannerDatabase
from conversion import columns_to_native_numpy, prepare_data_for_graphing

def main():
    db = SpannerDatabase("gvizlab", "vizlab", "graphcast")

    mock_db = MockSpannerDatabase()
    mock_query_result, mock_fields = mock_db.execute_query("")
    mock_d, mock_ignored_columns = columns_to_native_numpy(mock_query_result, mock_fields)
    mock_graph = prepare_data_for_graphing(mock_d, False)

    simple_query = "GRAPH FinGraph MATCH (p:Person)-[:OWNS]->(acc:Account) RETURN TO_JSON(p) as person_json_single, TO_JSON(acc) as account_json_single;"
    simple_query_result, simple_fields = db.execute_query(simple_query)
    simple_d, simple_ignored_columns = columns_to_native_numpy(simple_query_result, simple_fields)
    simple_graph = prepare_data_for_graphing(simple_d, False)

    complex_query = "GRAPH FinGraph MATCH  (synthesized_id:Person {id: 5})-[:OWNS]->(:Account)  (()-[transfer:Transfers]->(layering_accnt)){1,4}  (:Account)<-[:OWNS]-(another_synthesized_id:Person {id: 9}) RETURN TO_JSON(layering_accnt) AS layering_accnts, TO_JSON(transfer) AS transfers LIMIT 5"
    complex_query_result, complex_fields = db.execute_query(complex_query)
    complex_d, complex_gnored_columns = columns_to_native_numpy(complex_query_result, complex_fields)
    complex_graph = prepare_data_for_graphing(complex_d, False)

if __name__ == "__main__":
    main()