diff --git a/relational/querysplit.py b/relational/querysplit.py index 72cfa58..a80d07a 100644 --- a/relational/querysplit.py +++ b/relational/querysplit.py @@ -17,3 +17,26 @@ # author Salvo "LtWorf" Tomaselli # # This module splits a query into a program. + + +def vargen(avoid, prefix=''): + ''' + Generates temp variables. + + Avoid contains variable names to skip. + ''' + count = 0 + + while True: + r = '' + c = count + while True: + r = chr((c % 26) + 97) + r + if c < 26: + break + c //= 26 + + r = prefix + r + if r not in avoid: + yield r + count += 1