"""Generates an ORM class for the given table"""

def __init__(self, sql_table, pkg):

self.sql_table = sql_table

self.pkg = pkg

def __repr__(self):

java_cols = map(JavaColumn, self.sql_table._columns)

content_value_mapping = []

for i, java_col in enumerate(java_cols):

cursor_get = java_col.cursor_get

content_value_mapping.append("this.{} = {};".format(java_col.var_name,

cursor_get.format(i)))

return CLASS_TEMPLATE.format(

table=self.sql_table,

pkg=self.pkg,

sqltable='"\n+"'.join(str(self.sql_table).split('\n')),

classname=self.classname,

baseurihash=self.baseurihash,

baseitemhash=self.baseitemhash,

column_constants="\n ".join([x.declare_const for x in java_cols]),

column_constants_list=", ".join([x.const_name for x in java_cols]),

column_vars="\n ".join([x.declare_var for x in java_cols]),

column_field_from_cursor="\n ".join(content_value_mapping),

to_content_values=self.to_content_values)

@property

def to_content_values(self):

java_cols = map(JavaColumn, self.sql_table._columns)

no_id = []

for x in java_cols:

if x.var_name != "_id":

no_id.append(x)

result = ""

sep = "\n "

for java_col in no_id:

if 'NOT NULL' in java_col.column.constraint:

simple = True

else:

simple = False

if "CURRENT_" in java_col.column.constraint:

# Timestamp, special case here

result = \

sep.join([result,

"if ({1} != null)\

values.put({0}, {1});".format(java_col.const_name,

java_col.var_name)])

elif simple:

result = \

sep.join([result,

"values.put({0}, {1});".format(java_col.const_name,

java_col.var_name)])

else:

result = \

sep.join([result,

"if ({1} != null) {{\n\

}}".format(java_col.const_name,

java_col.var_name)])

return result

@property

def classname(self):

return "{0.name}Item".format(self.sql_table)

@property

def baseurihash(self):

s = sha1(self.classname)

s.update("baseuri")

return "0x" + s.hexdigest()[:7]

@property

def baseitemhash(self):

s = sha1(self.classname)

s.update("baseitem")

def __init__(self, sql_column):

self.column = sql_column

@property

def declare_const(self):

return COL_CONST_TEMPLATE.format(self.const_name,

self.var_name)

@property

def var_name(self):

return self.column.name

@property

def const_name(self):

name = self.column.upper_name

while name.startswith("_"):

name = name[1:]

return "COL_" + name

@property

def java_type(self):

st = self.column.type

if 'NOT NULL' in self.column.constraint:

simple = True

else:

simple = False

if st == "INTEGER":

return "long" if simple else "Long"

elif st == "REAL":

return "float" if simple else "Float"

elif st == "TIMESTAMP":

return "String"

else:

return "String"

@property

def cursor_get(self):

# Double braces are converted to single braces at the end of

# function.

base = "cursor.get{0}({{0}})"

if ('NOT NULL' not in self.column.constraint

and self.column.name != "_id"):

base = "cursor.isNull({{0}}) ? null : cursor.get{0}({{0}})"

st = self.column.type

if st == "INTEGER":

return base.format("Long")

elif st == "REAL":

return base.format("Float")

elif st == "TIMESTAMP":

return base.format("String")

else:

return base.format("String")

@property

def default_value(self):

if "PRIMARY KEY" in self.column.constraint:

return "= -1" #_id columns should have a non-null invalid value

elif "CURRENT_" in self.column.constraint:

return "= null"

elif "DEFAULT" in self.column.constraint:

val = self.column.constraint\

.replace('PRIMARY KEY', '')\

.replace('NOT NULL', '')\

.replace('DEFAULT', '', 1)\

.replace("'", '"').strip()

return "= {}".format(val)

else:

# No need to define a default value

return ""

@property

def declare_var(self):

return "public {0.java_type} {0.var_name} {0.default_value}"\