U - i @s@ddlmZmZddlmZddlmZddlmZddZdS))requestjsonify)closing)connect_to_database)datetimec Cst}|d}t}zz.t|jdd}|d|f|}|stt ddddfW5QRWW0S|d|f|}|st d ddd fW5QRWWS|d d kr|d |dd}nt |d|d }|d |}|d|||f|d|df| t d|dddfW5QRWW`SQRXWnPt k r} z0| t dt| dddfWYWSd} ~ XYnXW5|XdS)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_jsongetrclosercursorexecutefetchonermincommit Exceptionrollbackstr) invoice_iddatar connectionrinvoiceZdiscountZdiscount_amountrer!8/var/www/html/api-medvista/Services/Billing/Discounts.pyapply_discountsJ &$   :r#N) flaskrr contextlibrconfrrr#r!r!r!r"s