from email import header


#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@Contact :   liuyuqi.gov@msn.cn
@Time    :   2022/11/26 19:59:03
@License :   Copyright © 2017-2022 liuyuqi. All Rights Reserved.
@Desc    :   获取手机号，手机验证码
'''
import json,re,sys,logging,requests
from twitter_account_creator.models import api
class Phone(object):

    def __init__(self, apiKey):
        pass

    def getBalance(self):
        res = json.loads(requests.get(api.getBalance).text)
        if res["response"] == "error":
            return {"balance": res["error_msg"]}
        if res["response"] == "1":
            return {"balance": res["balance"]}
        return {"balance": "something went wrong :/"}

    def getNumber(self):
        try:
            res = json.loads(requests.get(api.getNumber).text)
            number = res["number"]
            _id = res["id"]
            return number, _id
        except:
            return "error_number", "error_id"
    def getSMS(self, id):
        try:
            res = json.loads(requests.get(api.getSms).text)
            if res["sms"] != None:
                return res["sms"]
        except:
            return "error_code"