feat: add table neighbors

bd_api/apps/api/v1/models.py

@@ -981,6 +981,10 @@ def gbq_slug(self):
  table = cloud_table.gcp_table_id
  return f"basedosdados.{dataset}.{table}"
 
+ @property
+ def get_graphql_gbq_slug(self):
+ return self.gbq_slug
+
  @property
  def gcs_slug(self):
  """Get the slug used in Google Cloud Storage"""
@@ -989,10 +993,6 @@ def gcs_slug(self):
  table = cloud_table.gcp_table_id
  return f"staging/{dataset}/{table}"
 
- @property
- def get_graphql_gbq_slug(self):
- return self.gbq_slug
-
  @property
  def get_graphql_gcs_slug(self):
  return self.gcs_slug
@@ -1109,6 +1109,22 @@ def full_coverage(self) -> str:
  def get_graphql_full_coverage(self):
  return self.full_coverage
 
+ @property
+ def neighbors(self):
+ """Similiar tables and columns
+ - Tables and columns with similar directories
+ - Tables and columns with similar coverages or tags (WIP)
+ """
+ all_neighbors = []
+ for column in self.columns:
+ for neighbor in column.neighbors:
+ all_neighbors.append((neighbor.table, neighbor))
+ return all_neighbors
+
+ @property
+ def get_graphql_neighbors(self):
+ return self.neighbors
+
  def clean(self):
  """
  Clean method for Table model
@@ -1302,6 +1318,20 @@ def full_coverage(self) -> str:
  def get_graphql_full_coverage(self):
  return self.full_coverage
 
+ @property
+ def neighbors(self):
+ """Similiar tables and columns
+ - Tables and columns with similar directories
+ - Tables and columns with similar coverages or tags (WIP)
+ """
+ if not self.directory_primary_key:
+ return []
+ return Column.objects.filter(directory_primary_key=self.directory_primary_key).all()
+
+ @property
+ def get_graphql_neighbors(self):
+ return self.neighbors
+
 
 class ColumnOriginalName(BaseModel):
  """Model definition for ColumnOriginalName."""