def importData(rdb, inputfile):
sTables = [
Config, Subject, ProductGroups, Products, BankNames, BankAccounts
]
cTables = [CustGroups, Customers]
dTables = [Bill, Notebook, Factors, FactorItems, Cheque, ChequeHistory]
if rdb == "rdbClean":
return
if rdb == 'rdbS': # structure
tables = sTables
elif rdb == 'rdbSC':
tables = sTables + cTables
elif rdb == 'rdbAll':
tables = sTables + cTables + dTables
newdb = database.Database(inputfile, share.config.db_repository,
share.config.echodbresult)
for clas in tables:
newdb.session.query(clas).delete()
movingData = share.config.db.session.query(clas).all()
clas2 = clas.__table__
columns = list(clas2.columns.keys())
for d in movingData:
data = dict([(str(column), getattr(d, column))
for column in columns])
instant = clas(**data)
newdb.session.add(instant)
def checkInputDb(inputfile, selectedFormat):
import os
import sys
import logging
from sqlalchemy import create_engine
from sqlalchemy import MetaData, Table, Column, ForeignKey, ColumnDefault
from sqlalchemy import exc
dbtypes = (
("sqlite", "sqlite:///"), ("sql", "mysql://")
) # mysql won't be used in this function. It is just for filling dbTypes !
filename = ""
if selectedFormat != 1:
# filename = ( 'directory' , 'file.format')
filename = os.path.split(inputfile)
splitByDot = filename[1].split(".")
l = len(splitByDot)
if l > 1: # if file name is with format (e.g .sqlite)
filetype = splitByDot[l - 1]
if filetype == dbtypes[selectedFormat][0]:
type = dbtypes[selectedFormat][1]
filename = filename[1]
else:
return -1
else: # if filename is without any format
filetype = dbtypes[selectedFormat][0]
type = dbtypes[selectedFormat][1]
inputfile += "." + filetype
filename = filename[1] + "." + filetype
# if not os.path.isfile(inputfile):
# return filename
try:
#engine = create_engine(type + inputfile, echo=True)
database.Database(inputfile, share.config.db_repository,
share.config.echodbresult)
print(" yesss")
except exc.DatabaseError:
logging.debug(sys.exc_info()[0])
return -2
return filename
def createDb(dbName, builder):
# creating new empty db
from gi.repository import Gtk
pathname = os.path.join(os.path.dirname(amirconfig.__file__),
amirconfig.__amir_data_directory__)
abs_data_path = os.path.abspath(pathname)
db_repository = os.path.join(abs_data_path, 'amir_migrate')
dbformat = "sqlite"
dbType = "sqlite:///"
from platform import system
fileaddress = os.path.join(share.config.confdir, dbName + "." + dbformat)
dbFile = dbType + fileaddress
try:
newdb = database.Database(dbFile, db_repository,
share.config.echodbresult)
except Exception as e:
msg = _(
"There was a problem in creating new database. Maybe trying with another name will help...\n"
+ str(e))
msgbox = Gtk.MessageDialog(builder.get_object("window1"),
Gtk.DialogFlags.MODAL,
Gtk.MessageType.ERROR, Gtk.ButtonsType.OK,
msg)
msgbox.set_title(_("Error creating new database"))
msgbox.run()
msgbox.destroy()
return
if builder.get_object("radiobutton2").get_active():
################# filling db with importing ###########
e = {
"chkDoc": (Bill, Notebook),
"chkBank": (BankNames, BankAccounts),
"chkConf": (Config, ),
"chkNote": (Subject, ),
"chkFact": (Factors, FactorItems),
"chkUser": (),
"chkProductG": (ProductGroups, ),
"chkProduct": (Products, ),
"chkCust": (Customers, ),
"chkCheq": (Cheque, )
}
checkboxes = builder.get_object("grid5").get_children()
dbClasses = [] # permissions ...
for checkbox in checkboxes:
if checkbox.get_active():
a = Gtk.Buildable.get_name(checkbox) # widget ID
for clas in e[str(a)]:
dbClasses.append(clas)
for clas in dbClasses:
newdb.session.query(clas).delete()
movingData = share.config.db.session.query(clas).all()
clas2 = clas.__table__
columns = list(clas2.columns.keys())
for d in movingData:
data = dict([(str(column), getattr(d, column))
for column in columns])
instant = clas(**data)
newdb.session.add(instant)
# # or :
# moved = (clas2.delete())
# newdb.session.execute(moved)
# print [c.name for c in clas2.columns]
# insert = clas2.insert().from_select( [c.name for c in clas2.columns],clas2.select()) #select([c.name for c in config.db.session.clas]) )
# newdb.session.execute(insert)
newdb.session.commit()