def __init__(self, timeframe, table, name):
self.sql = None
field = Field(name, table=table)
if timeframe == "date*":
alias = "{}_date".format(field.get_sql().replace("_", ""))
self.sql = fn.Date(field, alias=alias)
elif timeframe == "month*":
alias = "{}_month".format(field.get_sql().replace("_", ""))
self.sql = fn.ToChar(DateTrunc("month", field), "YYYY-MM", alias=alias)
elif timeframe == "week*":
alias = "{}_week".format(field.get_sql().replace("_", ""))
self.sql = fn.ToChar(DateTrunc("week", field), "YYYY-MM-DD", alias=alias)
elif timeframe == "year*":
alias = "{}_year".format(field.get_sql().replace("_", ""))
self.sql = fn.Extract("YEAR", field, alias=alias)
def test__tochar__(self):
q = Q.from_(self.t).select(fn.ToChar(self.t.foo, "SomeFormat"))
self.assertEqual("SELECT TO_CHAR(\"foo\",'SomeFormat') FROM \"abc\"",
str(q))
def test__tochar__(self):
q = Q.from_(self.t).select(fn.ToChar(self.t.foo, "SomeFormat"))
self.assertEqual('SELECT TO_CHAR("foo",\'SomeFormat\') FROM "abc"',
str(q))
