40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
from argparse import ArgumentDefaultsHelpFormatter, ArgumentParser
|
|
from proteus import Model
|
|
from proteus import config as pconfig
|
|
|
|
from tools import open_file
|
|
|
|
import json
|
|
|
|
dir = "./demo/data"
|
|
file_ = "parties.csv"
|
|
path = "".join([dir, "/", file_])
|
|
|
|
|
|
def set_config(database, config_file):
|
|
return pconfig.set_trytond(database, config_file=config_file)
|
|
|
|
|
|
def setup_parties(database, config_file):
|
|
set_config(database, config_file)
|
|
data = open_file(path)
|
|
|
|
Party = Model.get("party.party")
|
|
parties = [
|
|
Party(name=r.get("Name")) for r in data]
|
|
|
|
return Party.save(parties)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
with open('demo/data/config.json') as file:
|
|
config_tryton = json.load(file)
|
|
|
|
parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
|
|
parser.add_argument('-c', '--config', dest='config_file')
|
|
parser.add_argument('-d', '--database', dest='database',
|
|
default='demo', help="database name")
|
|
options = parser.parse_args()
|
|
|
|
setup_parties(options.database, options.config_file)
|