mirror of
https://github.com/c0de-archive/Odoo-Contact-Import.git
synced 2025-01-01 20:42:39 +00:00
Initial Commit
First commit on this repo, first test too
This commit is contained in:
commit
8df6301f64
17
README.md
Normal file
17
README.md
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
ERP Contact Import Tool
|
||||||
|
=======================
|
||||||
|
|
||||||
|
This utility fixes the contacts csv file for importing into the ERP
|
||||||
|
Specifically this utility will replace all the values in a csv column with a dictionary created through another csv file
|
||||||
|
|
||||||
|
Copyright (c) 2016 David Todd (alopexc0de) https://c0defox.es
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in
|
||||||
|
all copies or substantial portions of the Software.
|
91
main.py
Normal file
91
main.py
Normal file
@ -0,0 +1,91 @@
|
|||||||
|
# ERP Contact Import Tool
|
||||||
|
# This utility fixes the contacts csv file for importing into the ERP
|
||||||
|
# Specifically this utility is for the step of replacing the company names with their export_ids after the companies have been imported
|
||||||
|
|
||||||
|
# Copyright (c) 2016 David Todd (alopexc0de) https://c0defox.es
|
||||||
|
|
||||||
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
# of this software and associated documentation files (the "Software"), to deal
|
||||||
|
# in the Software without restriction, including without limitation the rights
|
||||||
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
# copies of the Software, and to permit persons to whom the Software is
|
||||||
|
# furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
# The above copyright notice and this permission notice shall be included in
|
||||||
|
# all copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
import csv
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
|
||||||
|
# Change these paths to reflect your needs
|
||||||
|
config = {
|
||||||
|
'partner_path': 'C:\\Users\\dtodd\\Desktop\\res.partner (2).csv', # This is the csv file containing two columns, id and name
|
||||||
|
'import_path': 'C:\\Users\\dtodd\\Desktop\\To Import\\MN Companies-50 FTE-HQ one (contacts) Campaign.csv', # This is the csv file containing whatever needs to be imported
|
||||||
|
'temp_path': 'C:\\Users\\dtodd\\Desktop\\temp', # This is a csv file that will be deleted on program completion
|
||||||
|
'column_to_change': 'parent_id/id' # This is the name of the column that you want to change
|
||||||
|
}
|
||||||
|
|
||||||
|
# Find the index number of a column in a row and return that
|
||||||
|
def find_index(row=None, col=None):
|
||||||
|
if ((row == None) | (col == None)) | ((type(row) != '' | type(col) != ''):
|
||||||
|
raise ValueError('find_index requires both row and col to be set to strings')
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
index = None
|
||||||
|
for i,j in enumerate(row):
|
||||||
|
if j == col:
|
||||||
|
index = i
|
||||||
|
break
|
||||||
|
return index
|
||||||
|
|
||||||
|
os.rename(config['import_path'], config['import_path']+'.orig') # Store the original import_path in a backup file
|
||||||
|
|
||||||
|
with open(config['partner_path'], 'r') as partner_csv:
|
||||||
|
reader = csv.reader(partner_csv)
|
||||||
|
|
||||||
|
first_row = next(reader)
|
||||||
|
if ('id' not in first_row) | ('name' not in first_row):
|
||||||
|
raise IndexError('Either \'id\' or \'name\' is not defined in your partner_path csv file')
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
id = find_index(first_row, 'id')
|
||||||
|
name = find_index(first_row, 'name')
|
||||||
|
|
||||||
|
if (id == None) | (name == None):
|
||||||
|
raise ValueError('Either id or name could not be indexed with find_index')
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
partner_ids = {rows[id]:rows[name] for rows in reader} # Build a dict containing the company names as the keys and the __export__IDs as values
|
||||||
|
reader = None
|
||||||
|
|
||||||
|
# Open both the import_csv and temp_csv files
|
||||||
|
with open(config['import_path'], 'r') as import_csv, open(config['temp_path'], 'w') as temp_csv:
|
||||||
|
reader = csv.reader(import_csv)
|
||||||
|
writer = csv.writer(temp_csv)
|
||||||
|
|
||||||
|
first_row = next(reader)
|
||||||
|
if config['column_to_change'] not in first_row:
|
||||||
|
raise IndexError('\'{}\' is not defined in your import_path csv file').format(config['column_to_change'])
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
col = find_index(first_row, config['column_to_change'])
|
||||||
|
|
||||||
|
if col != None:
|
||||||
|
for lines in reader:
|
||||||
|
# Taken from http://stackoverflow.com/a/2400577/5727514 and modified to support what I need
|
||||||
|
pattern = re.compile('|'.join(partner_ids.keys()))
|
||||||
|
lines[col] = pattern.sub(lambda x: partner_ids[x.group()], lines[col])
|
||||||
|
|
||||||
|
writer.writerow(lines)
|
||||||
|
else:
|
||||||
|
raise ValueError('col could not be indexed with find_index')
|
||||||
|
exit(1)
|
||||||
|
|
||||||
|
# Close our files to be nice to the OS
|
||||||
|
temp_csv.close()
|
||||||
|
import_csv.close()
|
||||||
|
partner_csv.close()
|
||||||
|
|
||||||
|
# On program completion, move the temp file to the original file
|
||||||
|
os.rename(config['temp_path'], config['import_path'])
|
Loading…
Reference in New Issue
Block a user