Adding Vars trick

terminusdb · Sep 2, 2022 · ac2224f · ac2224f
1 parent 31261fd
commit ac2224f
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 4 deletions.
diff --git a/terminusdb_client/__init__.py b/terminusdb_client/__init__.py
@@ -1,6 +1,6 @@
 from .client import Patch, Client # noqa
 from .woqldataframe import woqlDataframe as WOQLDataFrame # noqa
-from .woqlquery import WOQLQuery  # noqa
+from .woqlquery import WOQLQuery, Var, Vars # noqa
 from .woqlschema import * # noqa
 # Backwards compatibility
 WOQLClient = Client # noqa

diff --git a/terminusdb_client/query_syntax/query_syntax.py b/terminusdb_client/query_syntax/query_syntax.py
@@ -1,11 +1,11 @@
-from ..woqlquery import WOQLQuery, Var, Doc # noqa
+from ..woqlquery import WOQLQuery, Var, Vars, Doc # noqa
 import re
 import sys
 
 __BARRED = ['re', 'vars']
 __ALLOWED = ['__and__', '__or__', '__add__']
 __module = sys.modules[__name__]
-__exported = ['Var', 'Doc']
+__exported = ['Var', 'Vars', 'Doc']
 
 
 def __create_a_function(attribute):

diff --git a/terminusdb_client/woqlquery/__init__.py b/terminusdb_client/woqlquery/__init__.py
@@ -1 +1 @@
-from .woql_query import WOQLQuery, Var, Doc # noqa
+from .woql_query import WOQLQuery, Var, Vars, Doc # noqa
diff --git a/terminusdb_client/woqlquery/woql_query.py b/terminusdb_client/woqlquery/woql_query.py
@@ -50,6 +50,12 @@ def __str__(self):
  return self.name
 
 
+class Vars:
+ def __init__(self, *args):
+ for arg in args:
+ setattr(self, arg, Var(arg))
+
+
 class Doc:
  def __init__(self, dictionary):
  self.dictionary = dictionary