def add_user(self, user_id, referrer_id=None):
with self.connection:
# --- Пользователь зарегистрировался по реферальной ссылке ---
if referrer_id != None:
return self.cursor.execute("INSERT INTO 'users' ('user_id', 'referrer_id') VALUES (?,?)", (user_id, referrer_id))
# --- Пользователь зарегистрировался не по реферальной ссылке ---
else:
return self.cursor.execute("INSERT INTO 'users' ('user_id') VALUES (?)", (user_id,))