Subversion Repositories SmartDukaan

'''

Created on 29-Mar-2010

@author: ashish

'''

from elixir import session

from shop2020.model.v1.order.impl import DataService, RedExpressUpdateService, \

    RechargeService

from shop2020.model.v1.order.impl.Convertors import to_t_transaction, to_t_alert, \

    to_t_order, to_t_lineitem, to_t_payment_settlement, to_t_verification_agent, \

    to_t_attribute, to_t_rechargeTransaction, to_t_frc, to_t_hotspot, \

    to_t_sourcedetail, to_t_amazonorder, to_t_storeOrderDetail, \

    to_t_storeOrderCollection, to_t_hotspotServiceMatrix, to_t_ebayOrder, \

    to_t_amazonFbaSalesSnapshot, to_t_snapdealOrder, to_t_flipkartOrder, \

    to_t_amazonFbaOrderReturn, to_t_flipkartAdvantageOrder, to_t_hsOrder, \

    to_t_Creditor, to_t_CreditHistory, to_t_UserSanction, to_t_LoanHistory, \

    to_t_PaginatedCreditHistory, to_t_PaginatedLoanHistory, \

    to_t_PaginatedUserSanction, to_t_returnOrderInfo, to_t_returnTransaction, \

    to_t_rPickupRequest, to_t_OutstandingPayments, to_t_ShipmentLogisticsCostDetail, \

    to_t_pmsa

from shop2020.model.v1.order.impl.DataAccessors import create_transaction, \

    get_transactions_for_customer, get_transaction_status, get_line_items_for_order, \

    get_transaction, get_transactions_for_shopping_cart_id, \

    change_transaction_status, get_orders_for_customer, get_orders_for_transaction, \

    get_order, get_returnable_orders_for_customer, \

    get_cancellable_orders_for_customer, get_orders_by_billing_date, get_all_orders, \

    change_order_status, get_alerts, add_alert, add_billing_details, close_session, \

    mark_orders_as_picked_up, mark_orders_as_delivered, mark_orders_as_rto, \

    order_outofstock, batch_orders, update_non_delivery_reason, \

    enqueue_transaction_info_email, get_undelivered_orders, get_order_for_customer, \

    get_valid_order_count, get_cust_count_with_successful_txn, \

    get_valid_orders_amount_range, get_valid_orders, toggle_doa_flag, \

    request_pickup_number, authorize_pickup, receive_return, validate_doa, \

    reship_order, refund_order, get_return_orders, process_return, get_return_order, \

    mark_doas_as_picked_up, verify_order, is_alive, get_orders_by_shipping_date, \

    update_weight, change_warehouse, change_product, add_delay_reason, \

    reconcile_cod_collection, get_transactions_requiring_extra_processing, \

    mark_transaction_as_processed, get_item_wise_risky_orders_count, \

    accept_order_for_item, get_orders_in_batch, get_order_count, \

    get_ebs_settlement_summaries, mark_order_cancellation_request_received, \

    mark_order_cancellation_request_denied, refund_transaction, \

    mark_order_cancellation_request_confirmed, \

    mark_transaction_as_payment_flag_removed, mark_orders_as_po_raised, \

    mark_orders_as_reversal_initiated, mark_orders_as_not_available, \

    update_shipment_address, mark_orders_as_timeout, get_order_for_awb, \

    mark_orders_as_shipped_from_warehouse, mark_alerts_as_seen, \

    mark_order_doa_request_received, mark_order_doa_request_authorized, \

    mark_order_return_request_received, mark_order_return_request_authorized, \

    add_invoice_number, get_reshipped_order_ids, validate_return_product, \

    get_orders_for_provider_for_status, save_ebs_settlement_summary, \

    get_ebs_settlement_date, save_payment_settlements, mark_ebs_settlement_uploaded, \

    get_billed_orders_for_vendor, get_settlement_for_Prepaid, \

    get_settlements_by_date, get_slipped_sipping_date_orders, \

    mark_order_as_lost_in_transit, get_cancelled_orders, mark_order_as_delivered, \

    mark_return_orders_as_picked_up, update_orders_as_PORaised, \

    get_order_distribution_by_status, get_orders_not_met_expected_delivery_date, \

    mark_orders_as_local_connected, mark_orders_as_destinationCityReached, \

    mark_orders_as_firstDeliveryAttempted, get_non_delivered_orders_by_courier, \

    get_orders_not_local_connected, get_doas_not_picked_up, \

    get_return_orders_not_picked_up, get_orders_not_picked_up, get_rto_orders, \

    get_order_list, get_order_ids_for_status, update_orders_as_paid_to_vendor, \

    update_COD_agent, get_refunded_orders_marked_paid, get_settlement_for_Cod, \

    get_order_list_for_vendor, update_order_only_as_paid_to_vendor, \

    get_all_verification_agents, get_all_attributes_for_order_id, \

    set_order_attribute_for_transaction, get_billed_orders, get_all_return_orders, \

    mark_order_as_received_at_store, get_receive_pending_orders, \

    get_received_at_store_orders, mark_orders_as_returned_from_store, \

    set_order_attributes, get_orders_collection_at_store, get_order_attribute_value, \

    change_jacket_number, mark_order_as_rto_in_transit, get_recharge_order, \

    get_recharge_orders, update_recharge_order_status, activate_recharge_txn, \

    get_user_wallet, get_user_wallet_history, get_service_providers, \

    get_service_provider_for_device, get_recharge_orders_for_transaction, \

    get_recharge_orders_for_device, get_recharge_orders_for_status, \

    get_plans_for_operator, get_recharge_statistics, validate_recharge, \

    get_recharge_denominations, update_availability_status, get_misc_charges, \

    get_available_emi_schemes, refund_recharge_order, create_recharge_transaction, \

    get_physical_orders, get_document, retrieve_invoice, change_shipping_address, \

    get_recharge_transactions, get_recharge_transaction, get_frcs, get_hotspot_store, \

    get_circle, retrieve_hotspot_recharge_invoice, get_recharge_trans, \

    split_freebie_order, get_recharge_transactions_by_number, \

    update_hotspot_store_password, get_source_detail, get_all_circles, delete_frcs, \

    topup_company_wallet, get_wallet_balance_for_company, add_amazon_order, \

    update_amazon_order_status, get_amazon_order, get_amazon_orders_shipped, \

    get_amazon_orders_cancelled, get_orders_for_store, \

    get_store_order_advance_invoice, add_frc, add_series, get_store_order_detail, \

    save_store_order_detail, get_all_edc_banks, save_refund_amounts_for_store_order, \

    get_collections_for_store, get_amazon_orders_to_acknowledge, \

    change_amazon_order_status, update_timestamp_for_amazon_order, \

    update_source_detail_timestamp, get_amazon_order_by_amazonorderid, \

    get_orders_by_mobile_number, get_orders_by_amazon_id, change_courier_provider, \

    convert_store_to_normal, update_freebie_item, update_order_AWB, \

    get_hotspot_service_matrices, get_orders_by_vendor, \

    get_orders_count_created_after_timestamp_for_source, \

    get_min_created_timestamp_undelivered_orders_for_source, \

    get_recharge_order_status, get_recharge_transaction_status, \

    accept_package_orders, get_group_orders_by_logistics_txn_id, \

    add_billing_details_for_groupped_orders, get_invoice_format_logistics_txn_id, \

    split_bulk_order, move_orders_to_correct_warehouse, \

    verify_orders_for_transaction, get_creditor_info, update_creditor_info, \

    get_user_sanction_details, update_user_sanction, get_credit_history_records, \

    process_credit_transaction, get_loan_payable_for_user_to_creditor, \

    get_loan_history_records, process_loan_transaction, \

    get_credit_history_records_paginated, get_loan_history_records_paginated, \

    get_user_sanction_details_paginated, get_orders_in_batch_as_promised_shipping, \

    set_order_attribute_for_master_order_id, update_master_order_awb, \

    add_or_update_shipment_logistics_cost_details, get_return_order_info, \

    get_return_orders_info_map, update_return_order_info, \

    bulk_update_return_order_info, get_return_orders_as_per_warehouse, \

    create_return_transaction, get_return_transactions_for_customer, \

    get_return_transaction, change_return_transaction_status, \

    create_return_pickup_request, update_return_pickup_request, \

    get_return_orders_for_return_transaction, receive_return_pickup, \

    validate_return_pickup, process_return_pickup, mark_return_transaction_complete, \

    refund_return_transaction_payment, get_eligible_orders_for_return, \

    get_eligible_return_orders_for_pickup, validate_return_transaction, \

    get_return_pickup_request, mark_return_not_required_orders_as_processed, \

    get_all_return_orders_for_return_pickup_request, unaccept_order, \

    get_outstanding_payments, mark_payment_settled, get_seller_info, \

    get_warehouse_address, get_buyer_by_warehouse, add_shipment_delay, \

    refund_return_order, get_cost_detail_for_logistics_txn_id, \

    add_shipment_logistic_detail, create_payment, calculate_payment_amount, \

    add_amount_in_wallet, get_billed_orders_for_manifest_gen, register_rsa, \

    add_sales_associate, search_pmsa, get_pmsa_user, update_pmsa_user, \

    get_pending_pmsa, get_pmsa_users, get_stats_for_associates, credit_user_wallet, \

    is_private_deal_transaction, add_invoice_details_to_orders, \

    mark_order_for_registered_gst_invoice, is_shipment_cod, \

    get_intransit_orders_on_date, get_intransit_orders_on_date_by_item_id

from shop2020.model.v1.order.impl.DataService import DtrBatchCreditTracker

from shop2020.model.v1.order.impl.ObsoleteDataAccessors import create_ebay_order, \

    get_ebay_order_by_orderId, get_ebay_order_by_sales_rec_number, get_ebay_order, \

    update_ebay_order, ebay_order_exists, update_order_for_ebay, split_ebay_order, \

    add_or_update_amazon_fba_sales_snapshot, get_amazon_fba_sales_snapshot_for_days, \

    get_amazon_fba_sales_latest_snapshot_for_item, create_snapdeal_order, \

    get_snapdeal_order, snapdeal_order_exists, update_latest_fba_prices_for_item, \

    flipkart_order_exists, create_flipkart_order, get_flipkart_order, \

    get_flipkart_order_by_subOrderId, update_flipkart_order_dates_and_awb, \

    update_snapdeal_orders_status, update_flipkart_orders_status, \

    bulk_add_or_update_amazon_fba_sales_snapshot, get_created_orders_for_flipkart, \

    change_easyship_mfn_order_txn_status, update_amazon_fba_order_returns, \

    get_all_amazon_fba_order_returns, get_total_sale_returns_fba_skus_curent_time, \

    get_amazon_fba_sales_latest_snapshot_for_item_location_wise, \

    get_verification_pending_orders_fk, get_fa_order_by_fk_order_id, \

    get_all_fa_orders_list, add_update_fa_orders_bulk, flipkart_fa_order_exists, \

    create_homeshop_order, get_homeshop_order, homeshop_order_exists

from shop2020.model.v1.order.impl.model.DTHRechargeOrder import DTHRechargeOrder

from shop2020.model.v1.order.impl.model.DigitalTransaction import \

    DigitalTransaction

from shop2020.model.v1.order.impl.model.MobileRechargeOrder import \

    MobileRechargeOrder

from shop2020.model.v1.order.impl.model.UserWalletHistory import \

    UserWalletHistory

from shop2020.model.v1.order.script.RechargeUnknownReport import order_list

from shop2020.model.v1.user.impl.Converters import to_t_address

from shop2020.thriftpy.model.v1.order.ttypes import TransactionServiceException, \

    RechargeType, AmazonOrdersToAcknowledge, WalletReferenceType

from shop2020.utils.Utils import to_py_date, get_fdate_tdate, to_java_date

import datetime

import json

class OrderServiceHandler:

    def __init__(self, dbname='transaction', db_hostname='localhost'):

        """

        Constructor

        """

        DataService.initialize(dbname, db_hostname)

    def createTransaction(self, transaction):

        """

        Parameters:

         - transaction

        """

        try:

            return create_transaction(transaction)

        finally:

            close_session()

    def getTransaction(self, id):

        """

            Get transaction methods.

        Parameters:

         - id

        """

        try:

            transaction = get_transaction(id)

            return to_t_transaction(transaction)

        finally:

            close_session()

    def getTransactionsForCustomer(self, customerId, from_date, to_date, status):

        """

        Parameters:

         - customerId

         - from_date

         - to_date

         - status

        """

        try:

            current_from_date, current_to_date = get_fdate_tdate(from_date, to_date)

            transactions = get_transactions_for_customer(customerId, current_from_date, current_to_date, status)

            t_transaction = []

            for transaction in transactions:

                t_transaction.append(to_t_transaction(transaction))

            return t_transaction

        finally:

            close_session()

    def getTransactionsForShoppingCartId(self, shoppingCartId):

        """

        Parameters:

            - shoppingCartId

        """

        try:

            transactions = get_transactions_for_shopping_cart_id(shoppingCartId)

            t_transaction = []

            for transaction in transactions:

                t_transaction.append(to_t_transaction(transaction))

            return t_transaction

        finally:

            close_session()

    def getTransactionStatus(self, transactionId):

        """

        Parameters:

         - transactionId

        """

        try:

            return get_transaction_status(transactionId)

        finally:

            close_session()

    def changeTransactionStatus(self, transactionId, status, description, pickUp, orderType, source):

        """

        Parameters:

         - transactionId

         - status

         - description

         - selfPickup

        """

        try:

            return change_transaction_status(transactionId, status, description, pickUp, orderType, source)

        finally:

            close_session()

    def getOrdersForTransaction(self, transactionId, customerId):

        """

        Returns list of orders for given transaction Id. Also filters based on customer Id so that

        only user who owns the transaction can view its order details.

        Parameters:

         - transactionId

         - customerId

        """

        try:

            orders = get_orders_for_transaction(transactionId, customerId)

            return [to_t_order(order) for order in orders]    

        finally:

            close_session()

    def getAllOrders(self, statuses, from_date, to_date, warehouse_id):

        """

        Parameters:

         - status

         - from_date

         - to_date

         - warehouse_id

        """

        try:

            current_from_date, current_to_date = get_fdate_tdate(from_date, to_date)        

            orders = get_all_orders(statuses, current_from_date, current_to_date, warehouse_id)

            return [to_t_order(order) for order in orders]

        finally:

            close_session()

    def getOrdersInBatch(self, statuses, offset, limit, warehouse_id, source):

        """

        Returns at most 'limit' orders with the given statuses for the given warehouse starting from the given offset.

        Pass the status as null and the limit as 0 to ignore them.

        Parameters:

         - statuses

         - offset

         - limit

         - warehouse_id

        """

        try:

            orders = get_orders_in_batch(statuses, offset, limit, warehouse_id, source)

            return [to_t_order(order) for order in orders]

        finally:

            close_session()

    def getOrderCount(self, statuses, warehouseId, source):

        """

        Returns the count of orders with the given statuses assigned to the given warehouse.

        Parameters:

         - statuses

         - warehouseId

        """

        try:

            return get_order_count(statuses, warehouseId, source)

        finally:

            close_session()

    def getOrdersByBillingDate(self, status, start_billing_date, end_billing_date, warehouse_id):

        """

        Parameters:

         - status

         - start_billing_date

         - end_billing_date

         - warehouse_id

        """

        try:

            current_start_billing_date, current_end_billing_date = get_fdate_tdate(start_billing_date, end_billing_date)

            orders = get_orders_by_billing_date(status, current_start_billing_date, current_end_billing_date, warehouse_id)

            return [to_t_order(order) for order in orders]

        finally:

            close_session()

    def getOrdersByShippingDate(self, fromShippingDate, toShippingDate, providerId, warehouseId, cod):

        """

        Returns orders for a particular provider and warehouse which were shipped between the given dates.

        Returned orders comprise of COD orders if cod parameter is true. It comprises of prepaid orders otherwise.

        Pass providerId and warehouseId as -1 to ignore both these parameters.

        Parameters:

         - fromShippingDate

         - toShippingDate

         - providerId

         - warehouseId

         - cod

        """

        try:

            from_shipping_date, to_shipping_date = get_fdate_tdate(fromShippingDate, toShippingDate)

            orders = get_orders_by_shipping_date(from_shipping_date, to_shipping_date, providerId, warehouseId, cod)

            return [to_t_order(order) for order in orders]

        finally:

            close_session()

    def getReturnableOrdersForCustomer(self, customerId, limit = None):

        """

        Parameters:

         - customerId

         - limit

        """

        try:

            return get_returnable_orders_for_customer(customerId, limit)

        finally:

            close_session()

    def getCancellableOrdersForCustomer(self, customerId, limit = None):

        """

        Parameters:

         - customerId

         - limit

        """

        try:

            return get_cancellable_orders_for_customer(customerId, limit)

        finally:

            close_session()

    def changeOrderStatus(self, orderId, status, description):

        """

        Parameters:

         - orderId

         - status

         - description

         - 

        """

        try:

            return change_order_status(self, orderId, status, description)

        finally:

            close_session()

    def verifyOrder(self, orderId):

        """

        Marks the given order as SUBMITTED_FOR_PROCESSING and updates the verified

        timestamp. It is intended to be used for COD orders but can be harmlessly

        used for all other orders as well.

        Throws an exception if no such order exists.

        Parameters:

         - orderId

        """

        try:

            return verify_order(orderId)

        finally:

            close_session()

    def acceptOrder(self, orderId):

        """

        Marks the given order as ACCEPTED and updates the accepted timestamp. If the

        given order is not a COD order, it also captures the payment if the same has

        not been captured.

        Throws an exception if no such order exists.

        Parameters:

         - orderId

        """

        try:

            return accept_package_orders([orderId])

        finally:

            close_session()

    def unacceptOrder(self, orderId):

        """

        Rollsback to Processing State

        Parameters:

         - orderId

        """

        try:

            return unaccept_order(orderId)

        finally:

            close_session()

    def getOrdersForCustomer(self, customerId, from_date, to_date, statuses):

        """

        Returns list of orders for the given customer created between the given dates and having the given statuses.

        Pass and empty list to ignore filtering on statuses.

        Parameters:

         - customerId

         - from_date

         - to_date

         - statuses

        """

        try:

            current_from_date, current_to_date = get_fdate_tdate(from_date, to_date)

            orders = get_orders_for_customer(customerId, current_from_date, current_to_date, statuses)

            return [to_t_order(order) for order in orders]

        finally:

            close_session()

    def getOrderForCustomer(self, orderId, customerId):

        """

        Returns an order for the order Id. Also checks if the order belongs to the customer whose Id is passed.

        Throws exception if either order Id is invalid or order does not below to the customer whose Id is passed.

        Parameters:

         - customerId

         - orderId

        """

        try:

            return to_t_order(get_order_for_customer(orderId, customerId))

        finally:

            close_session()

    def getOrder(self, id):

        """

        Parameters:

         - id

        """

        try:

            return to_t_order(get_order(id))

        finally:

            close_session()

    def getOrderList(self, order_ids):

        """

        Parameters:

         - order_ids

        """

        try:

            orders = get_order_list(order_ids)

            return [to_t_order(order) for order in orders]

        finally:

            close_session()

    def getOrderListForVendor(self, order_ids, vendorId):

        """

        Parameters:

         - order_ids

         - vendorId

        """

        try:

            orders = get_order_list_for_vendor(order_ids, vendorId)

            return [to_t_order(order) for order in orders]

        finally:

            close_session()

    def getLineItemsForOrder(self, orderId):

        """

        Parameters:

         - orderId

        """

        try:

            lineitems = get_line_items_for_order(orderId)

            t_lineitems = []

            for lineitem in lineitems:

                t_lineitems.append(to_t_lineitem(lineitem))

            return t_lineitems

        finally:

            close_session()

    def addBillingDetails(self, orderId, invoice_number, serialNumber, itemNumber, freebieWarehouseId, billedBy, jacketNumber, billingType, vendorId, authorize):

        """

        Adds jacket number and serial no. to the order. Doesn't update the serial no. if a -1 is supplied.

        Also marks the order as billed and sets the billing timestamp.

        Return false if it doesn't find the order with the given ID.

        Parameters:

         - orderId

         - jacketNumber

         - serialNumber

         - itemNumber

         - freebieWarehouseId

         - billedBy

        """

        try:

            return add_billing_details(orderId, invoice_number, serialNumber, itemNumber, freebieWarehouseId, billedBy, jacketNumber, billingType, vendorId, authorize)

        finally:

            close_session()

    def addInvoiceNumber(self, orderId, invoiceNumber, color, serialNumber, itemNumber):

        """

        Add the invoice number to the order.

        Parameters:

         - orderId

         - invoice_number

        """

        try:

            add_invoice_number(orderId, invoiceNumber, color, serialNumber, itemNumber)

        finally:

            close_session()

    def batchOrders(self, warehouseId):

        """

        Create a batch of all the pending orders for the given warehouse.

        The returned list is orderd by created_timestamp.

        If there are no pending orders, an empty list is returned.

        Parameters:

         - warehouseId

        """

        try:

            pending_orders = batch_orders(warehouseId)

            return [to_t_order(order) for order in pending_orders]

        finally:

            close_session()

    def markOrderAsOutOfStock(self, orderId):

        """

        Mark the given order as out of stock. Throws an exception if the order with the given Id couldn't be found.

        Parameters:

         - orderId

        """

        try:

            return order_outofstock(orderId)

        finally:

            close_session()

    def markOrdersAsShippedFromWarehouse(self, warehouseId, providerId, cod, orderIds):

        """

        Depending on the third parameter, marks either all prepaid or all cod orders BILLED by the

        given warehouse and were picked up by the given provider as SHIPPED_FROM_WH.

        Parameters:

         - warehouseId

         - providerId

         - cod

        """

        try:

            return mark_orders_as_shipped_from_warehouse(warehouseId, providerId, cod, orderIds)

        finally:

            close_session()

    def markOrdersAsReturnedFromStore(self, providerId, orderIds, awbs):

        """

        Parameters:

         - providerId

         - orderIds

         - awbs

        """

        try:

            return mark_orders_as_returned_from_store(providerId, orderIds, awbs)

        finally:

            close_session()

    def setOrderAttributes(self, orderId, attributes):

        """

        sets attributes for an order

        Parameters:

         - orderId

         - attributes

        """

        try:

            return set_order_attributes(orderId, attributes)

        finally:

            close_session()

    def markOrdersAsPickedUp(self, providerId, pickupDetails):

        """

        Marks all SHIPPED_FROM_WH orders of the previous day for a provider as SHIPPED_TO_LOGISTICS.

        Raises an exception if we encounter report for an AWB number that we did not ship.

        Parameters:

         - providerId

         - pickupDetails

        """

        try:

            mark_orders_as_picked_up(providerId, pickupDetails)

        finally:

            close_session()

    def getOrdersNotPickedUp(self, providerId):

        """

        Returns a list of orders that were shipped from warehouse but did not appear in the pick-up report.

        Parameters:

         - providerId

        """

        try:

            orders_not_picked_up = get_orders_not_picked_up(providerId)

            return [to_t_order(order) for order in orders_not_picked_up]

        finally:

            close_session()

    def markOrdersAsDelivered(self, providerId, deliveredOrders):

        """

        Marks all orders with AWBs in the given map as delivered. Also sets the delivery timestamp and

        the name of the receiver.

        Raises an exception if we encounter report for an AWB number that we did not ship.

        Parameters:

         - providerId

         - deliveredOrders

        """

        try:

            mark_orders_as_delivered(providerId, deliveredOrders)

        finally:

            close_session()

    def markOrderAsDelivered(self, orderId, deliveryTimestamp, receiver):

        """

        Attributes:

         - orderId

         - deliveryTimestamp

         - receiver

        """

        try:

            mark_order_as_delivered(orderId, to_py_date(deliveryTimestamp), receiver)

        finally:

            close_session()

    def markOrderAsReceivedAtStore(self, orderId, deliveryTimestamp):

        """

        Attributes:

         - orderId

         - deliveryTimestamp

        """

        try:

            mark_order_as_received_at_store(orderId, to_py_date(deliveryTimestamp))

        finally:

            close_session()

    def getReceivePendingOrders(self, storeId):

        """

        Attributes:

         - storeId

        """

        try:

            orders = get_receive_pending_orders(storeId)

            return [to_t_order(order) for order in orders]

        finally:

            close_session()

    def getReceivedAtStoreOrders(self, storeId):

        """

        Attributes:

         - storeId

        """

        try:

            orders = get_received_at_store_orders(storeId)

            return [to_t_order(order) for order in orders]

        finally:

            close_session()

    def getOrdersCollectionAtStore(self, storeId, fromDate, toDate, onlyCod):

        """

        Parameters:

         - storeId

         - fromDate

         - toDate

         - onlyCod

        """

        try:

            orders = get_orders_collection_at_store(storeId, to_py_date(fromDate), to_py_date(toDate), onlyCod)

            return [to_t_order(order) for order in orders]

        finally:

            close_session()

    def markAsRTOrders(self, providerId, returnedOrders):

        """

        Mark all orders with AWBs in the given map as RTO. Also sets the delivery timestamp.

        Raises an exception if we encounter report for an AWB number that we did not ship.

        Parameters:

         - providerId

         - returnedOrders

        """

        try:

            mark_orders_as_rto(providerId, returnedOrders)

        finally:

            close_session()

    def getRTOrders(self, providerId):

        """

        Returns a list of orders that were returned by courier.

        Parameters:

         - providerId

        """

        try:

            rto_orders = get_rto_orders(providerId)

            return [to_t_order(order) for order in rto_orders]

        finally:

            close_session()

    def updateNonDeliveryReason(self, providerId, undeliveredOrders):

        """

        Update the status description of orders whose AWB numbers are keys of the Map.

        Parameters:

         - providerId

         - undelivered_orders

        """

        try:

            update_non_delivery_reason(providerId, undeliveredOrders)

        finally:

            close_session()

    def getNonDeliveredOrdersbyCourier(self, providerId):

        """

        Returns a list of orders that were picked up or shipped four days ago but did not get delivered.

        Parameters:

         - providerId

        """

        try:

            orders_not_delivered = get_non_delivered_orders_by_courier(providerId)

            return [to_t_order(order) for order in orders_not_delivered]

        finally:

            close_session()

    def markOrdersAsLocalConnected(self, providerId, local_connected_orders):

        """

        Mark all orders with AWBs in the given map as local connected. Also sets the local connected timestamp.

        Parameters:

         - providerId

         - local_connected_orders

        """

        try:

            mark_orders_as_local_connected(providerId, local_connected_orders)

        finally:

            close_session()

    def getOrdersNotLocalConnected(self, providerId):

        """

        Returns a list of orders that were picked up or shipped but pending local connection.

        Parameters:

         - providerId

        """

        try:

            orders_pending_local_connection = get_orders_not_local_connected(providerId)

            return [to_t_order(order) for order in orders_pending_local_connection]

        finally:

            close_session()

    def markOrdersAsDestinationCityReached(self, providerId, destination_city_reached_orders):

        """

        Mark all orders with AWBs in the given map as reached destination city. Also sets the reached destination timestamp.

        Parameters:

         - providerId

         - destination_city_reached_orders

        """

        try:

            mark_orders_as_destinationCityReached(providerId, destination_city_reached_orders)

        finally:

            close_session()

    def markOrdersAsFirstDeliveryAttempted(self, providerId, first_atdl_orders):

        """

        Mark all orders with AWBs in the given map as first delivery attempt made. Also sets the first delivery attempted timestamp.

        Parameters:

         - providerId

         - first_atdl_orders

        """

        try:

            mark_orders_as_firstDeliveryAttempted(providerId, first_atdl_orders)

        finally:

            close_session()

    def getUndeliveredOrdersExpectedDeliveryDateNotMet(self):

        """

        Returns the list of orders whose expected delivery date has passed but have not been

        delivered yet.

        Returns an empty list if no such orders exist.

        """

        try:

            orders_not_delivered = get_orders_not_met_expected_delivery_date()

            return [to_t_order(order) for order in orders_not_delivered]

        finally:

            close_session()

    def getUndeliveredOrders(self, providerId, warehouseId):

        """

        Returns the list of orders whose delivery time has passed but have not been

        delivered yet for the given provider and warehouse. To get a complete list of

        undelivered orders, pass them as -1.

        Returns an empty list if no such orders exist.

        Parameters:

         - providerId

         - warehouseId

        """

        try:

            undelivered_orders = get_undelivered_orders(providerId, warehouseId)

            return [to_t_order(order) for order in undelivered_orders]

        finally:

            close_session()

    def enqueueTransactionInfoEmail(self, transactionId):

        """

        Save the email containing order details to be sent to customer later by batch job

        Parameters:

         - transactionId

        """

        try:

            return enqueue_transaction_info_email(transactionId)

        finally:

            close_session()

    def getAlerts(self, type, warehouseId, status, timestamp):

        """

        Parameters:

         - type

         - warehouseId

         - status

         - timestamp

        """

        try:

            alerts = get_alerts(type, warehouseId, status, timestamp)

            t_alerts = []

            for alert in alerts:

                t_alerts.append(to_t_alert(alert)) 

            return t_alerts

        finally:

            close_session()

    def addAlert(self, type, warehouseId, description):

        """

        Parameters:

         - type

         - description

         - warehouseId

        """

        try:

            add_alert(type, warehouseId, description)

        finally:

            close_session()

    def markAlertsAsSeen(self, warehouseId):

        """

        Parameters:

         - warehouseId

        """

        try:

            mark_alerts_as_seen(warehouseId)

        finally:

            close_session()

        pass

    def getValidOrderCount(self, ):

        """

        Return the number of valid orders. (OrderStatus >= OrderStatus.SUBMITTED_FOR_PROCESSING)

        """

        try:

            return get_valid_order_count()

        finally:

            close_session()

    def getNoOfCustomersWithSuccessfulTransaction(self, ):

        """

        Returns the number of distinct customers who have done successful transactions

        """

        try:

            return get_cust_count_with_successful_txn()

        finally:

            close_session()

    def getValidOrdersAmountRange(self, ):

        """

        Returns the minimum and maximum amounts of a valid order. (OrderStatus >= OrderStatus.SUBMITTED_FOR_PROCESSING)

        List contains two values, first minimum amount and second maximum amount.

        """

        try:

            return get_valid_orders_amount_range()

        finally:

            close_session()

    def getValidOrders(self, limit, onlyStore):

        """

        Returns list of Orders in descending order by Order creation date. List is restricted to limit Orders.

        If limit is passed as 0, then all valid Orders are returned.

        Parameters:

         - limit

         - onlyStore

        """

        try:

            return [to_t_order(order) for order in get_valid_orders(limit, onlyStore)]

        finally:

            close_session()

    def toggleDOAFlag(self, orderId):

        """

        Toggle the DOA flag of an order. This should be used to flag an order for follow-up and unflag it when the follow-up is complete.

        Returns the final flag status.

        Throws an exception if the order with the given id couldn't be found or if the order status is not DELVIERY_SUCCESS.

        Parameters:

         - orderId

        """

        try:

            return toggle_doa_flag(orderId)

        finally:

            close_session()

    def markOrderDoaRequestReceived(self, orderId):

        """

        Once user raise the request for a DOA, order status will be changed from DELVIERY_SUCCESS to DOA_REQUEST_RECEIVED

        Parameters:

         - orderId

        """

        try:

            return mark_order_doa_request_received(orderId)

        finally:

            close_session()

    def markOrderDoaRequestAuthorized(self, orderId, isAuthorized,  fromStore, isReship):

        """

        CRM person can authorize or deny the request reised by customer. If he authorizes order will change from DOA_REQUEST_RECEIVED

        to DOA_REQUEST_AUTHORIZED. If he denies, status will be changed back to DELVIERY_SUCCESS.

        Parameters: