o i @s@ddlmZmZddlmZddlmZddlmZddZdS))requestjsonify)closing)connect_to_database)datetimec Cst}|d}t}zzt|jdd}|d|f|}|s:tddddfWdWW| S|d|f|}|s]td ddd fWdWW| S|d d krn|d |dd}n t |d|d }|d |}|d|||f|d|df| td|dddfWdWW| S1swYWn)t y} z| tdt| dddfWYd} ~ W| Sd} ~ wwW| dS| w)N discount_codeT) dictionaryz:SELECT * FROM invoices WHERE id = %s AND status = 'unpaid'zUnpaid invoice not found)messageerroria8 SELECT * FROM discounts WHERE code = %s AND is_active = TRUE AND (start_date <= NOW() OR start_date IS NULL) AND (end_date >= NOW() OR end_date IS NULL) AND (max_uses IS NULL OR used_count < max_uses) zInvalid discount codeiZ discount_typeZ percentage total_amountamountdz UPDATE invoices SET discount_amount = %s, total_amount = %s WHERE id = %s z~ UPDATE discounts SET used_count = used_count + 1 WHERE id = %s idzDiscount applied successfullyF)r new_totalr zError applying discount: i)rget_jsongetrrcursorexecutefetchonerclosemincommit Exceptionrollbackstr) invoice_iddatar connectionrinvoiceZdiscountZdiscount_amountrer!T/Users/harjaapdhillon/Desktop/newidea/medvista_backend/Services/Billing/Discounts.pyapply_discounts`  6 6   62( 6r#N) flaskrr contextlibrconfrrr#r!r!r!r"s