예제 #1
0
    def post(self):
        current_user = get_jwt_identity()
        user = User.find_by_email(current_user)
        data = parser.parse_args()
        tax_info = Tax.find_by_userid(userid=user.id, year=data.get('year'))
        if tax_info is None:
            tax_info = Tax(user_id=user.id, year=data.get('year'))
            db.session.add(tax_info)
            db.session.commit()
        try:
            tax_info.collected_at_source = data.get('collected_at_source')
            tax_info.advance_paid = data.get('advance_paid')
            tax_info.adjustment = data.get('adjustment')
            tax_info.total_tax = data.get('total_tax')

            db.session.commit()

            return make_response(
                jsonify({
                    'status': 200,
                    'msg': "Tax Information Updated"
                }), 200)
        except Exception:
            return make_response(
                jsonify({
                    'status': 500,
                    'msg': "Internal server error"
                }), 500)
예제 #2
0
 def get(self):
     current_user = get_jwt_identity()
     user = User.find_by_email(current_user)
     data_ = get_parser.parse_args()
     tax_info = Tax.find_by_userid(userid=user.id, year=data_.get('year'))
     try:
         tax_schema = TaxSchema()
         data = tax_schema.dump(tax_info)
         return make_response(
             jsonify({
                 'status': 200,
                 'data': data,
                 'msg': "Tax Information"
             }), 200)
     except Exception:
         return make_response(
             jsonify({
                 'status': 500,
                 'msg': "Internal server error"
             }), 500)