在Python中，确定字符串是电子邮件还是整数的最快方法是什么？

Question

我希望能够使用提供的电子邮件地址或用户 ID（整数）从数据库中提取用户。为此，我必须检测提供的字符串是整数还是电子邮件。寻找最快的方法来做到这一点。谢谢。

def __init__(self, data):
    #populate class data
    self._fetchInfo(data)


def _fetchInfo(self, data):
    #If an email
        #SELECT ... WHERE email = 'data'
    #or if a user_id
        #SELECT ... WHERE id = 'data'

    #Fill class attributes 
    self._id = row['id']
    self._email = row['id']
    ...

Answer 1

在 Python 中处理这个问题的规范方法是先尝试，再请求宽恕：

def _fetchInfo(self, data):
    try:
        data=int(data)
        sql='SELECT ... WHERE id = %s'
        args=[data]
    except ValueError:
        sql='SELECT ... WHERE email = %s'
        args=[data]
        # This might fail, in which case, data was neither a valid integer or email address

这个策略也被称为 “请求宽恕比请求许可更容易”。

Answer 2

您可以使用 isinstance 函数：

if isinstance(data, int):
   # it's an id
else:
   # it's a string

虽然就个人而言，我只有两个方法、fetchById 和 fetchByEmail 来明确其工作原理。

Answer 3

你说两者都是字符串，对吗？这也行。

if data.isdigit():
    # it's an id
else:
    # it's not

Answer 4

if '@' in data:
    # email
else:
    # id