В python, какой самый быстрый способ определить, является ли строка электронным письмом или целым числом?

Question

Я бы хотел иметь возможность извлекать пользователей из базы данных, используя предоставленный адрес электронной почты или идентификатор пользователя (целое число). Чтобы сделать это, я должен определить, является ли предоставленная строка целым числом или электронным письмом. Ищете самый быстрый способ сделать это. Спасибо.

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

Вы сказали, что оба были струнами, верно? Это тоже сработает.

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

Answer 3

Вы можете использовать функцию isinstance :

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

Хотя лично у меня было бы два метода, fetchById и fetchByEmail, чтобы было понятно, как это работает.

Answer 4

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