_inherit = "shipping.move"

_columns = {

'shipment_identific_no': fields.char('ShipmentIdentificationNumber', size=64,),

'logo': fields.binary('Logo'),

'tracking_url': fields.char('Tracking URL', size=512,),

'service': fields.many2one('ups.shipping.service.type', 'Shipping Service'),

'shipper': fields.many2one('ups.account.shipping', 'Shipper', help='The specific user ID and shipper. Setup in the company configuration.'),

}

def print_label(self, cr, uid, ids, context=None):

if not ids: return []

return {

'type': 'ir.actions.report.xml',

'report_name': 'ship.log.label.print',

'datas': {

'model':'shipping.move',

'id': ids and ids[0] or False,

'ids': ids and ids or [],

'report_type': 'pdf'

},

'nodestroy': True

}

def getTrackingUrl(self, cr, uid, ids, context=None):

ship_log_obj = self.browse(cr, uid, ids[0], context=context)

if ship_log_obj.tracking_no:

tracking_url = "http://wwwapps.ups.com/WebTracking/processInputRequest?sort_by=status&tracknums_displayed=1&\

_inherit = "stock.picking"

def _get_company_code(self, cr, user, context=None):

res = super(stock_picking, self)._get_company_code(cr, user, context=context)

res.append(('ups', 'UPS'))

return list(set(res))

_columns = {

'ups_service': fields.many2one('ups.shipping.service.type', 'Service', help='The specific shipping service offered'),

'shipper': fields.many2one('ups.account.shipping', 'Shipper', help='The specific user ID and shipper. Setup in the company configuration.'),

'shipment_digest': fields.text('ShipmentDigest'),

'negotiated_rates': fields.float('NegotiatedRates'),

'shipment_identific_no': fields.char('ShipmentIdentificationNumber', size=64,),

'tracking_no': fields.char('TrackingNumber', size=64,),

'trade_mark': fields.related('shipper', 'trademark', type='char', size=1024, string='Trademark'),

'ship_company_code': fields.selection(_get_company_code, 'Ship Company', method=True, size=64),

'ups_pickup_type': fields.selection([

('01', 'Daily Pickup'),

('03', 'Customer Counter'),

('06', 'One Time Pickup'),

('07', 'On Call Air'),

('11', 'Suggested Retail Rates'),

('19', 'Letter Center'),

('20', 'Air Service Center'),

], 'Pickup Type'),

'ups_packaging_type': fields.many2one('shipping.package.type', 'Packaging Type'),

'ups_use_cc': fields.boolean('Credit Card Payment'),

'ups_cc_type': fields.selection([

('01', 'American Express'),

('03', 'Discover'),

('04', 'MasterCard'),

('05', 'Optima'),

('06', 'VISA'),

('07', 'Bravo'),

('08', 'Diners Club')

], 'Card Type'),

'ups_cc_number': fields.char('Credit Card Number', size=32),

'ups_cc_expiaration_date': fields.char('Expiaration Date', size=6, help="Format is 'MMYYYY'"),

'ups_cc_security_code': fields.char('Security Code', size=4,),

'ups_cc_address_id': fields.many2one('res.partner', 'Address'),

'ups_third_party_account': fields.char('Third Party Account Number', size=32),

'ups_third_party_address_id': fields.many2one('res.partner', 'Third Party Address'),

'ups_third_party_type': fields.selection([('shipper', 'Shipper'), ('consignee', 'Consignee')], 'Third Party Type'),

'ups_bill_receiver_account': fields.char('Receiver Account', size=32, help="The UPS account number of Freight Collect"),

'ups_bill_receiver_address_id': fields.many2one('res.partner', 'Receiver Address')

}

def on_change_sale_id(self, cr, uid, ids, sale_id=False, state=False, context=None):

vals = {}

if sale_id:

sale_obj = self.pool.get('sale.order').browse(cr, uid, sale_id)

service_type_obj = self.pool.get('ups.shipping.service.type')

ups_shipping_service_ids = service_type_obj.search(cr, uid, [('description', '=', sale_obj.ship_method_id.name)], context=context)

if ups_shipping_service_ids:

vals['ups_service'] = ups_shipping_service_ids[0]

shipping_obj = self.pool.get('ups.account.shipping')

ups_shipping_ids = shipping_obj.search(cr, uid, [('ups_shipping_service_ids', 'in', ups_shipping_service_ids[0])], context=context)

if ups_shipping_ids:

vals['shipper'] = ups_shipping_ids[0]

log_company_obj = self.pool.get('logistic.company')

logistic_company_ids = log_company_obj.search(cr, uid, [('ups_account_shipping_ids', 'in', ups_shipping_ids[0])], context=context)

if logistic_company_ids:

vals['logis_company'] = logistic_company_ids[0]

return {'value': vals}

def onchange_bill_shipping(self, cr, uid, ids, bill_shipping, ups_use_cc, ups_cc_address_id, ups_bill_receiver_address_id, partner_id,

shipper, context=None):

vals = {}

if bill_shipping == 'shipper':

if not ups_cc_address_id and shipper:

ship_address = self.pool.get('ups.account.shipping').read(cr, uid, shipper, ['address'], context=context)['address']

if ship_address:

vals['ups_cc_address_id'] = ship_address[0]

else:

vals['ups_use_cc'] = False

if not ups_bill_receiver_address_id:

vals['ups_bill_receiver_address_id'] = partner_id

_inherit = "stock.picking.out"

def _get_company_code(self, cr, user, context=None):

return self.pool.get('stock.picking')._get_company_code(cr, user, context)

_columns = {

'ups_service': fields.many2one('ups.shipping.service.type', 'Service', help='The specific shipping service offered'),

'shipper': fields.many2one('ups.account.shipping', 'Shipper', help='The specific user ID and shipper. Setup in the company configuration.'),

'shipment_digest': fields.text('ShipmentDigest'),

'negotiated_rates': fields.float('NegotiatedRates'),

'shipment_identific_no': fields.char('ShipmentIdentificationNumber', size=64,),

'tracking_no': fields.char('TrackingNumber', size=64,),

'trade_mark': fields.related('shipper', 'trademark', type='char', size=1024, string='Trademark'),

'ship_company_code': fields.selection(_get_company_code, 'Ship Company', method=True, size=64),

'ups_pickup_type': fields.selection([

('01', 'Daily Pickup'),

('03', 'Customer Counter'),

('06', 'One Time Pickup'),

('07', 'On Call Air'),

('11', 'Suggested Retail Rates'),

('19', 'Letter Center'),

('20', 'Air Service Center'),

], 'Pickup Type'),

'ups_packaging_type': fields.many2one('shipping.package.type', 'Packaging Type'),

'ups_use_cc': fields.boolean('Credit Card Payment'),

'ups_cc_type': fields.selection([

('01', 'American Express'),

('03', 'Discover'),

('04', 'MasterCard'),

('05', 'Optima'),

('06', 'VISA'),

('07', 'Bravo'),

('08', 'Diners Club')

], 'Card Type'),

'ups_cc_number': fields.char('Credit Card Number', size=32),

'ups_cc_expiaration_date': fields.char('Expiaration Date', size=6, help="Format is 'MMYYYY'"),

'ups_cc_security_code': fields.char('Security Code', size=4,),

'ups_cc_address_id': fields.many2one('res.partner', 'Address'),

'ups_third_party_account': fields.char('Third Party Account Number', size=32),

'ups_third_party_address_id': fields.many2one('res.partner', 'Third Party Address'),

'ups_third_party_type': fields.selection([('shipper', 'Shipper'), ('consignee', 'Consignee')], 'Third Party Type'),

'ups_bill_receiver_account': fields.char('Receiver Account', size=32, help="The UPS account number of Freight Collect"),

'ups_bill_receiver_address_id': fields.many2one('res.partner', 'Receiver Address')

}

def onchange_bill_shipping(self, cr, uid, ids, bill_shipping, ups_use_cc, ups_cc_address_id, ups_bill_receiver_address_id, partner_id,

shipper, context=None):

vals = {}

if bill_shipping == 'shipper':

if not ups_cc_address_id and shipper:

ship_address = self.pool.get('ups.account.shipping').read(cr, uid, shipper, ['address'], context=context)['address']

if ship_address:

vals['ups_cc_address_id'] = ship_address[0]

else:

vals['ups_use_cc'] = False

if not ups_bill_receiver_address_id:

vals['ups_bill_receiver_address_id'] = partner_id

return {'value' : vals}

# def action_process(self, cr, uid, ids, context=None):

# sale_order_line = []

# deliv_order = self.browse(cr, uid, ids, context=context)

# if isinstance(deliv_order, list):

# deliv_order = deliv_order[0]

# do_transaction = True

# sale = deliv_order.sale_id

# if sale and sale.payment_method == 'cc_pre_auth' and not sale.invoiced:

# rel_voucher = sale.rel_account_voucher_id

# rel_voucher_id = rel_voucher and rel_voucher.id or False

# if rel_voucher_id and rel_voucher.state != 'posted' and rel_voucher.cc_auth_code:

# do_transaction = False

# vals_vouch = {'cc_p_authorize': False, 'cc_charge': True}

# if 'trans_type' in rel_voucher._columns.keys():

# vals_vouch.update({'trans_type': 'AuthCapture'})

# self.pool.get('account.voucher').write(cr, uid, [rel_voucher_id], vals_vouch, context=context)

# do_transaction = self.pool.get('account.voucher').authorize(cr, uid, [rel_voucher_id], context=context)

# if not do_transaction:

# self.write(cr, uid, ids, {'ship_state': 'hold', 'ship_message': 'Unable to process creditcard payment.'})

# cr.commit()

# raise osv.except_osv(_('Final credit card charge cannot be completed!'), _("Please hold shipment and contact customer service.."))

# return super(stock_picking_out, self).action_process(cr, uid, ids, context=context)

def action_done(self, cr, uid, ids, context=None):

res = super(stock_picking_out, self).action_done(cr, uid, ids, context=context)

for picking in self.browse(cr, uid, ids, context=context):

vals = {}

service_type_obj = self.pool.get('ups.shipping.service.type')

ship_method_id = picking.sale_id and picking.sale_id.ship_method_id

if ship_method_id:

service_type_ids = service_type_obj.search(cr, uid, [('description', 'like', ship_method_id.name)], context=context)

if service_type_ids:

vals['ups_service'] = service_type_ids[0]

service_type = service_type_obj.browse(cr, uid, service_type_ids[0], context=context)

if service_type.ups_account_id:

vals['shipper'] = service_type.ups_account_id.id

if service_type.ups_account_id.logistic_company_id:

vals['logis_company'] = service_type_obj.ups_account_id.logistic_company_id.id

return True

def on_change_sale_id(self, cr, uid, ids, sale_id=False, state=False, context=None):

vals = {}

if sale_id:

sale_obj = self.pool.get('sale.order').browse(cr, uid, sale_id)

service_type_obj = self.pool.get('ups.shipping.service.type')

ups_shipping_service_ids = service_type_obj.search(cr, uid, [('description', '=', sale_obj.ship_method_id.name)], context=context)

if ups_shipping_service_ids:

vals['ups_service'] = ups_shipping_service_ids[0]

shipping_obj = self.pool.get('ups.account.shipping')

ups_shipping_ids = shipping_obj.search(cr, uid, [('ups_shipping_service_ids', 'in', ups_shipping_service_ids[0])], context=context)

if ups_shipping_ids:

vals['shipper'] = ups_shipping_ids[0]

log_company_obj = self.pool.get('logistic.company')

logistic_company_ids = log_company_obj.search(cr, uid, [('ups_account_shipping_ids', 'in', ups_shipping_ids[0])], context=context)

if logistic_company_ids:

vals['logis_company'] = logistic_company_ids[0]

return {'value': vals}

def fill_addr(self, addr_id):

ret = {

'AddressLine1': addr_id and addr_id.street or '',

'AddressLine2': addr_id and addr_id.street2 or '',

'AddressLine3': "",

'City': addr_id and addr_id.city or '',

'StateProvinceCode': addr_id and addr_id.state_id.id and addr_id.state_id.code or '',

'PostalCode': addr_id and addr_id.zip or '',

'CountryCode': addr_id and addr_id.country_id.id and addr_id.country_id.code or '',

'PostalCode': addr_id.zip or ''

}

if addr_id and addr_id.classification == '2':

ret.update({'ResidentialAddress': ""})

return ret

def create_ship_accept_request_new(self, cr, uid, do, context=None):

if not do.shipper:

return ''

xml_ship_accept_request = """<?xml version="1.0"?>

<AccessRequest xml:lang='en-US'>

<AccessLicenseNumber>%(access_l_no)s</AccessLicenseNumber>

<UserId>%(user_id)s</UserId>

<Password>%(password)s</Password>

</AccessRequest>

""" % {

'access_l_no': do.shipper.access_license or '',

'user_id': do.shipper.userid or '',

'password': do.shipper.password,

}

xml_ship_accept_request += """<?xml version="1.0"?>

<ShipmentAcceptRequest>

<Request>

<TransactionReference>

<CustomerContext>%(customer_context)s</CustomerContext>

<XpciVersion>1.0001</XpciVersion>

</TransactionReference>

<RequestAction>ShipAccept</RequestAction>

</Request>

<ShipmentDigest>%(shipment_digest)s</ShipmentDigest>

</ShipmentAcceptRequest>

""" % {

'customer_context': do.name or '',

'shipment_digest': do.shipment_digest,

}

return xml_ship_accept_request

#

def process_ship_accept(self, cr, uid, do, packages, context=None):

shipment_accept_request_xml = self.create_ship_accept_request_new(cr, uid, do, context=context)

if do.logis_company.test_mode:

acce_web = do.logis_company.ship_accpt_test_web or ''

acce_port = do.logis_company.ship_accpt_test_port

else:

acce_web = do.logis_company.ship_accpt_web or ''

acce_port = do.logis_company.ship_accpt_port

if acce_web:

parse_url = urlparse(acce_web)

serv = parse_url.netloc

serv_path = parse_url.path

else:

raise osv.except_osv(_('Unable to find Shipping URL!'), _("Please configure the shipping company with websites."))

conn = httplib.HTTPSConnection(serv, acce_port)

res = conn.request("POST", serv_path, shipment_accept_request_xml)

import xml2dic

res = conn.getresponse()

result = res.read()

response_dic = xml2dic.main(result)

NegotiatedRates = ''

ShipmentIdentificationNumber = ''

TrackingNumber = ''

label_image = ''

control_log_image = ''

status = 0

status_description = ''

for response in response_dic['ShipmentAcceptResponse']:

if response.get('Response'):

for resp_items in response['Response']:

if resp_items.get('ResponseStatusCode') and resp_items['ResponseStatusCode'] == '1':

status = 1

if resp_items.get('ResponseStatusDescription'):

status_description = resp_items['ResponseStatusDescription']

if resp_items.get('Error'):

for err in resp_items['Error']:

if err.get('ErrorSeverity'):

status_description += '\n' + err.get('ErrorSeverity')

if err.get('ErrorDescription'):

status_description += '\n' + err.get('ErrorDescription')

do.write({'ship_message': status_description})

packages_ids = [package.id for package in do.packages_ids]

if status:

shipment_identific_number, tracking_number_notes, ship_charge = '', '', 0.0

for shipmentresult in response_dic['ShipmentAcceptResponse']:

if shipmentresult.get('ShipmentResults'):

package_obj = self.pool.get('stock.packages')

for package in response['ShipmentResults']:

if package.get('ShipmentIdentificationNumber'):

shipment_identific_number = package['ShipmentIdentificationNumber']

continue

ship_charge += package.get('ShipmentCharges') and float(package['ShipmentCharges'][2]['TotalCharges'][1]['MonetaryValue']) or 0.0

if package.get('PackageResults'):

label_image = ''

tracking_number = ''

label_code = ''

tracking_url = do.logis_company.ship_tracking_url or ''

for tracks in package['PackageResults']:

if tracks.get('TrackingNumber'):

tracking_number = tracks['TrackingNumber']

if tracking_url:

try:

tracking_url = tracking_url % tracking_number

except Exception, e:

tracking_url = "Invalid tracking url on shipping company"

if tracks.get('LabelImage'):

for label in tracks['LabelImage']:

if label.get('LabelImageFormat'):

for format in label['LabelImageFormat']:

label_code = format.get('Code')

if label.get('GraphicImage'):

label_image = label['GraphicImage']

im_in_raw = base64.decodestring(label_image)

path = tempfile.mktemp('.txt')

temp = file(path, 'wb')

temp.write(im_in_raw)

result = base64.b64encode(im_in_raw)

(dirName, fileName) = os.path.split(path)

self.pool.get('ir.attachment').create(cr, uid,

{

'name': fileName,

'datas': result,

'datas_fname': fileName,

'res_model': self._name,

'res_id': do.id,

'type': 'binary'

},

context=context)

temp.close()

try:

new_im = Image.open(path)

new_im = new_im.rotate(270)

new_im.save(path, 'JPEG')

except Exception, e:

pass

label_from_file = open(path, 'rb')

label_image = base64.encodestring(label_from_file.read())

label_from_file.close()

if label_code == 'GIF':

package_obj.write(cr, uid, packages_ids[packages], {

'tracking_no': tracking_number,

'shipment_identific_no': shipment_identific_number,

'logo': label_image,

'ship_state': 'in_process',

'tracking_url': tracking_url,

'att_file_name': fileName

}, context=context)

else:

package_obj.write(cr, uid, packages_ids[packages], {

'tracking_no': tracking_number,

'shipment_identific_no': shipment_identific_number,

'ship_state': 'in_process',

'tracking_url': tracking_url,

'att_file_name': fileName

}, context=context)

if int(time.strftime("%w")) in range(1, 6) or (time.strftime("%w") == '6' and do.sat_delivery):

next_pic_date = time.strftime("%Y-%m-%d")

else:

timedelta = datetime.timedelta(7 - int(time.strftime("%w")))

next_pic_date = (datetime.datetime.today() + timedelta).strftime("%Y-%m-%d")

package_data = package_obj.read(cr, uid, packages_ids[packages], ['weight', 'description'], context=context)

# i += 1

ship_move_obj = self.pool.get('shipping.move')

if label_code == 'GIF':

ship_move_obj.create(cr, uid, {

'pick_id': do.id,

'package_weight': package_data['weight'],

'partner_id': do.partner_id.id,

'service': do.ups_service.id,

'ship_to': do.partner_id.id,

'ship_from': do.ship_from and do.ship_from_address.id or \

do.shipper and do.shipper.address and do.shipper.address.id,

'tracking_no': tracking_number,

'shipment_identific_no': shipment_identific_number,

'logo': label_image,

'state': 'ready_pick',

'tracking_url': tracking_url,

'package': package_data['description'] and str(package_data['description'])[:126],

'pic_date': next_pic_date,

'sale_id': do.sale_id.id and do.sale_id.id or False,

}, context=context)

else:

ship_move_obj.create(cr, uid, {

'pick_id': do.id,

'package_weight': package_data['weight'],

'partner_id': do.partner_id.id,

'service': do.ups_service.id,

'ship_to': do.partner_id.id,

'ship_from': do.ship_from and do.ship_from_address.id or \

do.shipper and do.shipper.address and do.shipper.address.id,

'tracking_no': tracking_number,

'shipment_identific_no': shipment_identific_number,

'state': 'ready_pick',

'tracking_url': tracking_url,

'package': package_data['description'] and str(package_data['description'])[:126],

'pic_date': next_pic_date,

'sale_id': do.sale_id.id and do.sale_id.id or False,

}, context=context)

tracking_number_notes += '\n' + tracking_number

if package.get('ControlLogReceipt'):

for items in package['ControlLogReceipt']:

if items.get('GraphicImage'):

control_log_image = items['GraphicImage']

im_in_raw = base64.decodestring(control_log_image)

file_name = tempfile.mktemp()

path = file_name = '.html'

temp = file(path, 'wb')

temp.write(im_in_raw)

temp.close()

label_from_file = open(path, 'rb')

control_log_image = base64.encodestring(label_from_file.read())

label_from_file.close()

package_obj.write(cr, uid, packages_ids, {'control_log_receipt': control_log_image, }, context=context)

do.write({'ship_state': 'ready_pick', 'ship_charge': ship_charge, 'internal_note': tracking_number_notes}, context=context)

return status, label_code

def add_product(self, cr, uid, package_obj):

prods = []

tot_weight = 0

for pkg in package_obj.pick_id.packages_ids:

tot_weight += pkg.weight

for move_lines in package_obj.pick_id.move_lines:

product_id = move_lines.product_id

if move_lines.product_id.supply_method == 'produce':

produce = "Yes"

else:

produce = "NO[1]"

product = {

'Description': move_lines.product_id.description or " ",

'Unit': {

'Number': str(int(move_lines.product_qty) or 0),

'Value': str((move_lines.product_id.list_price * move_lines.product_qty) or 0),

'UnitOfMeasurement': {'Code': "LBS", 'Description': "Pounds"}

},

'CommodityCode': package_obj.pick_id.comm_code or "",

'PartNumber': "",

'OriginCountryCode': package_obj.pick_id.address_id and package_obj.pick_id.address_id.country_id and \

package_obj.pick_id.address_id.country_id.code or "",

'JointProductionIndicator': "",

'NetCostCode': "NO",

'PreferenceCriteria': "B",

'ProducerInfo': produce,

'MarksAndNumbers': "",

'NumberOfPackagesPerCommodity': str(len(package_obj.pick_id.packages_ids)),

'ProductWeight': {

'UnitOfMeasurement': {'Code': "LBS", 'Description': "Pounds"},

'Weight': "%.1f" % (tot_weight or 0.0)

},

'VehicleID': "",

}

prods.append(product)

return prods

def create_comm_inv(self, cr, uid, package_obj):

invoice_id = False

if package_obj.pick_id.sale_id:

if package_obj.pick_id.sale_id.invoice_ids:

invoice_id = package_obj.pick_id.sale_id.invoice_ids[0]

user = self.pool.get('res.users').browse(cr, uid, uid)

comm_inv = {

'FormType': "01",

'Product': [], # Placed out of this dictionary for common use

'InvoiceNumber': "",

'InvoiceDate': "",

'PurchaseOrderNumber': "",

'TermsOfShipment': "",

'ReasonForExport': "SALE",

'Comments': "",

'DeclarationStatement': "I hereby certify that the good covered by this shipment qualifies as an originating good for purposes of \

preferential tariff treatment under the NAFTA.",

'CurrencyCode': user.company_id.currency_id.name or "",

}

if invoice_id:

comm_inv['InvoiceNumber'] = invoice_id.number or '/'

if invoice_id.date_invoice:

d = invoice_id.date_invoice

comm_inv['InvoiceDate'] = d[:4] + d[5:7] + d[8:10]

if not comm_inv['InvoiceDate']:

comm_inv['InvoiceDate'] = time.strftime("%Y%m%d")

return comm_inv

def create_cer_orig(self, cr, uid, package_obj):

cer_orig = {

'FormType': "03",

'Product': [], # Placed out of this dictionary for common use

'ExportDate': time.strftime("%Y%m%d"),

'ExportingCarrier': package_obj.pick_id.exp_carrier or "",

}

return cer_orig

def get_value(self, cr, uid, object, message=None, context=None):

if message is None:

message = {}

if message:

try:

from mako.template import Template as MakoTemplate

message = tools.ustr(message)

env = {

'user':self.pool.get('res.users').browse(cr, uid, uid, context=context),

'db':cr.dbname

}

templ = MakoTemplate(message, input_encoding='utf-8')

reply = MakoTemplate(message).render_unicode(object=object, peobject=object, env=env, format_exceptions=True)

return reply or False

except Exception:

logging.exception("Can't render %r", message)

return u""

else:

return message

def process_ship(self, cr, uid, ids, context=None):

company = self.pool.get('res.users').browse(cr, uid, uid, context=context).company_id

res = {

'type': 'ir.actions.client',

'tag': 'printer_proxy.print',

'name': _('Print Shipping Label'),

'params': {

'printer_name': company.printer_proxy_device_name,

'url': company.printer_proxy_url,

'username': company.printer_proxy_username,

'password': company.printer_proxy_password,

'data': [],

'format': 'epl2'

}

}

data = self.browse(cr, uid, type(ids) == type([]) and ids[0] or ids, context=context)

# Pass on up this function call if the shipping company specified was not UPS.

if data.ship_company_code != 'ups':

return super(stock_picking_out, self).process_ship(cr, uid, ids, context=context)

if not (data.logis_company or data.shipper or data.ups_service):

raise osv.except_osv("Warning", "Please select a Logistics Company, Shipper and Shipping Service.")

if not (data.logis_company and data.logis_company.ship_company_code == 'ups'):

return super(stock_picking_out, self).process_ship(cr, uid, ids, context=context)

if not data.packages_ids or len(data.packages_ids) == 0:

raise osv.except_osv("Warning", "Please define your packages.")

error = False

ups_config = api.v1.get_config(cr, uid, sale=data.sale_id, context=context)

for pkg in data.packages_ids:

try:

# Get the shipping label, store it, and return it.

label = api.v1.get_label(ups_config, data, pkg)

res['params']['data'].append(base64.b64encode(label.label))

self.pool.get('stock.packages').write(cr, uid, [pkg.id], {

'logo':label.label, 'tracking_no': label.tracking,

'ship_message': 'Shipment has processed'

})

except Exception, e:

if not error:

error = []

error_str = str(e)

error.append(error_str)

if error:

self.pool.get('stock.packages').write(cr, uid, pkg.id, {'ship_message': error_str}, context=context)

if not error:

self.write(cr, uid, data.id, {

'ship_state':'ready_pick', 'ship_message': 'Shipment has been processed.'

}, context=context)

return res

else:

self.write(cr, uid, data.id, {

'ship_message': 'Error occured on processing some of packages, ' +

'for details please see the status packages.'

}, context=context)

res = {

'type': 'ir.actions.client',

'tag': 'action_warn',

'name': 'Failure',

'params': {

'title': 'Package Errors',

'text': 'Errors encountered while processing packages. Look at package ship messages for details.',

'sticky': True

}

}

return res

def _get_journal_id(self, cr, uid, ids, context=None):

journal_obj = self.pool.get('account.journal')

vals = []

for pick in self.browse(cr, uid, ids, context=context):

src_usage = pick.move_lines[0].location_id.usage

dest_usage = pick.move_lines[0].location_dest_id.usage

type = pick.type

if type == 'out' and dest_usage == 'supplier':

journal_type = 'purchase_refund'

elif type == 'out' and dest_usage == 'customer':

journal_type = 'sale'

elif type == 'in' and src_usage == 'supplier':

journal_type = 'purchase'

elif type == 'in' and src_usage == 'customer':

journal_type = 'sale_refund'

else:

journal_type = 'sale'

value = journal_obj.search(cr, uid, [('type', '=', journal_type)], context=context)

for jr_type in journal_obj.browse(cr, uid, value, context=context):

t1 = jr_type.id, jr_type.name

if t1 not in vals:

vals.append(t1)

return vals

def do_partial(self, cr, uid, ids, partial_datas, context=None):

res = self._get_journal_id(cr, uid, ids, context=context)

result_partial = super(stock_picking_out, self).do_partial(cr, uid, ids, partial_datas, context=context)

if res and res[0]:

journal_id = res[0][0]

result = result_partial

for picking_obj in self.browse(cr, uid, ids, context=context):

sale = picking_obj.sale_id

if sale and sale.order_policy == 'picking':

pick_id = result_partial[picking_obj.id]['delivered_picking']

result = self.action_invoice_create(cr, uid, [pick_id], journal_id, type=None, context=context)

inv_obj = self.pool.get('account.invoice')

if result:

inv_obj.write(cr, uid, result.values(), {

'shipcharge': sale.shipcharge,

'sale_account_id': sale.ship_method_id and sale.ship_method_id.account_id and \

sale.ship_method_id.account_id.id or False,

'ship_method_id': sale.ship_method_id and sale.ship_method_id.id})

inv_obj.button_reset_taxes(cr, uid, result.values(), context=context)

_inherit = "stock.invoice.onshipping"

def create_invoice(self, cr, uid, ids, context=None):

if context is None:

context = {}

invoice_ids = []

res = super(stock, self).create_invoice(cr, uid, ids, context=context)

invoice_ids += res.values()

picking_pool = self.pool.get('stock.picking.out')

invoice_pool = self.pool.get('account.invoice')

active_picking = picking_pool.browse(cr, uid, context.get('active_id', False), context=context)

if active_picking:

invoice_pool.write(cr, uid, invoice_ids, {'shipcharge':active_picking.shipcharge }, context=context)

_inherit = "stock.move"

def created(self, cr, uid, vals, context=None):

if not context: context = {}

package_obj = self.pool.get('stock.packages')

pack_id = None

package_ids = package_obj.search(cr, uid, [('pick_id', "=", vals.get('picking_id'))])

if vals.get('picking_id'):

rec = self.pool.get('stock.picking').browse(cr, uid, vals.get('picking_id'), context)

if not context.get('copy'):

if not package_ids:

pack_id = package_obj.create(cr, uid , {'package_type': rec.sale_id.ups_packaging_type.id, 'pick_id': vals.get('picking_id')})

res = super(stock_move, self).create(cr, uid, vals, context)

if not context.get('copy'):

context.update({'copy': 1})

default_vals = {}

if pack_id:

default_vals = {'package_id':pack_id, 'picking_id':[]}

elif package_ids:

default_vals = {'package_id':package_ids[0], 'picking_id':[]}

self.copy(cr, uid, res, default_vals , context)