def load_data_file(self, files, loaded_tables=None):
data = ''
for i in files:
data += open(i).read()
data += '\n'
dict_data = {}
for i in data.split('\n'):
if i:
table_name, d_value = i.split('\t')
if table_name in dict_data:
dict_data[table_name].append(eval(d_value))
else:
dict_data[table_name] = [eval(d_value)]
rows_created = 0
for i in loaded_tables:
if i in dict_data:
table = orm.get_table(i)
for j in dict_data[i]:
table.create(**j).save()
rows_created += 1
update_sequence(table)
return rows_created
def dump_tables(self, tables):
import tempfile
try:
os.mkdir('/tmp/pyjitsu_sql_dump/')
except:
pass
filename = tempfile.mktemp()
print "Saving row data in %s." % filename
f = open(filename, 'w')
for i in tables:
tmp = orm.get_table(i).dump_row_values()
if tmp:
f.write(tmp + '\n')
f.flush()
f.close()
return filename
