fix onTable (#304)

itsumura-h · Jan 28, 2025 · 38c6b61 · 38c6b61
1 parent 6e5dd41
commit 38c6b61
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/allographer/schema_builder/models/column.nim b/src/allographer/schema_builder/models/column.nim
@@ -287,14 +287,19 @@ proc reference*(self:Column, column:string):Column =
 
 
 proc onTable*(self:Column, table:string):Column =
-  self.info["table"] = %*table
+  if self.info.kind == JNull:
+    self.info = %*{"table": table}
+  else:
+    self.info["table"] = %table
   return self
 
+
 when NimMajor == 1:
   proc on*(self:Column, table:string):Column {.deprecated: "Consider using onTable instead".} =
     self.info["table"] = %*table
     return self
 
+
 proc onDelete*(self:Column, kind:ForeignOnDelete):Column =
   self.foreignOnDelete = kind
   return self