from flask import Flask
from flask import requests, url_for
from payapp import PayApp

app = Flask(__name__)
pay = PayApp(PayApp.TEST_ID, PayApp.TEST_KEY, PayApp.TEST_VALUE)

@app.route('/')
def index():
    callback_url = url_for('.callback', _external=True)
    result = pay.pay_request(
        name=u'bery tasti fud',
        price='1000',
        contact='01000000000',
        returns='', # => 모바일에서 결제가 완료된 후 보여줄 페이지 URL
        callback=callback_url # => 백노티가 전달되는 URL
    )
    result.success # => 성공여부
    result.error # => 실패일 때 에러메시지
    result.identifier # => 결제 ID (payapp api의 mul_no)

    return result.url # => 결제 URL (모바일로 전송됨)

# callback url(payapp api의 feedback url)의 경우 특정 이벤트가 있을 때
# 백그라운드로 전달됩니다.
# 이 부분에서 state를 확인한 후 DB에 미리 state를 기록한 후에
# 프론트엔드에서 ajax 폴링/롱폴링 또는 '결제완료' 버튼을 통해
# 확인하도록 구현하면 됩니다.
@app.route('/callback', methods=['POST'])
def callback():
    data = request.form.to_dict()
    if not pay.verify_callback(data):  # 올바른 콜백 데이터인지 검증
        return 'FAIL'

    parsed = pay.parse_callback(data)  # 원하는 데이터로 가공
    parsed.identifier # => 결제 ID (payapp api의 mul_no)
    parsed.req_date # => python datetime 형식의 요청 날짜 및 시간
    parsed.pay_date # => python datetime 형식의 결제 날짜 및 시간 (결제 state일 때만 유효)
    parsed.pay_type # => payapp 문서 참고
    parsed.pay_state # => payapp 문서 참고

    # 신용카드 거래일 때 유효한 속성
    parsed.pay_check_url # => 카드전표 URL
    parsed.pay_card_name # => 결제 카드사 이름

    # 가상계좌 거래일 때 유효한 속성
    parsed.pay_vbank # => 가상계좌 은행 이름
    parsed.pay_vbank_account # => 가상계좌 계좌번호

    return 'SUCCESS'

@app.route('/cancel', methods=['GET'])
def cancel():
    identifier = request.args.get('id')
    memo = request.args.get('memo', '취소') # => 취소 사유 (문자로 전달됩니다)
    result = pay.pay_cancel(identifier=identifier, memo=memo)

    return result.success

if __name__ == '__main__':
    app.run(debug=True, threaded=True)