SpannerPropertyGraph/generate_data.py at main · maguec/SpannerPropertyGraph · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#! /usr/bin/env python

import os
from models.nodes import *
from models.edges import *
from models.utils import *
from models.embeddings import *
from google.cloud import spanner

if __name__ == "__main__":
    s = spanner.Client()
    spanner_insance_name = os.getenv("GOOGLE_CLOUD_SPANNER_INSTANCE", "properties")
    instance = s.instance(spanner_insance_name)
    client = instance.database("propertydb")
    owners = Owners(items=550)
    client.run_in_transaction(writeSpanner, owners)
    counties = Counties(items=60)
    client.run_in_transaction(writeSpanner, counties)
    properties = Properties(items=1000)
    client.run_in_transaction(writeSpanner, properties)
    embeddings = Description(items=len(properties.list_items))
    client.run_in_transaction(writeSpanner, embeddings)
    credit_reports = CreditReports(items=len(owners.list_items))
    client.run_in_transaction(writeSpanner, credit_reports)
    in_county = CountyEdges(properties, counties)
    client.run_in_transaction(writeSpanner, in_county)
    has_owner = PropertyEdges(properties, owners)
    client.run_in_transaction(writeSpanner, has_owner)
    has_credit = CreditEdges(owners, credit_reports)
    client.run_in_transaction(writeSpanner, has_credit)
    social = SocialEdges(owners)
    client.run_in_transaction(writeSpanner, social)
    companies = Companies(items=25)
    client.run_in_transaction(writeSpanner, companies)
    has_employee = CompanyEdges(companies, owners)
    client.run_in_transaction(writeSpanner, has_employee)
    embedding_edges = EmbedEdges(properties)
    client.run_in_transaction(writeSpanner, embedding_edges)