o HiP@sHddlmZddlmZddlZedZeejGdddZ dS))Database) PyMongoErrorN DBManagerc@sZeZdZdefddZddededefdd Zdd edefd d Z ddedefddZ dS)MongoDBManagerdbcCs||_tddS)NzUsing injected MongoDB instance)rloggerinfo)selfrr @/var/www/html/Trade-python/app/v1/services/zerodha/db_manager.py__init__ szMongoDBManager.__init__zerodha_portfoliosuser_idcollection_namereturnc Csrz|j|}|d|iddd}|r|dgWSgWSty8}ztd|gWYd}~Sd}~ww)Nrr)_idholdingsrzError fetching portfolio: )rfind_onegetrrerror)r rr collection portfolioer r r get_user_portfolio s z!MongoDBManager.get_user_portfoliotrading_signals signal_datac Csfz|j|}||}td|j|jWSty2}ztd|WYd}~dSd}~ww)NzSignal saved with ID: zError saving signal: )r insert_onerr inserted_idrr)r rrrresultrr r r save_signals  zMongoDBManager.save_signalorder_executions order_datac Csjz|j|}||}td|d|jWSty4}ztd|WYd}~dSd}~ww)NzExecution log saved for order: order_idzError saving execution log: )rrrrrrrr)r r"rrrrr r r save_execution_logs  z!MongoDBManager.save_execution_logN)r )r)r!) __name__ __module__ __qualname__rr strlistrdictr r$r r r r rs   r) pymongo.databaserpymongo.errorsrlogging getLoggerrsetLevelINFOrr r r r s