codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat 支持Emby/Jellyfin登录认证

Browse Source
This commit is contained in:
jxxghp
2023-06-20 13:09:11 +08:00
parent 47f5942e22
commit 0776a0b235
17 changed files with 117 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app/modules/emby/__init__.py
View File

@ -21,13 +21,14 @@ class EmbyModule(_ModuleBase):
def init_setting(self) -> Tuple[str, Union[str, bool]]:
return "MEDIASERVER", "emby"
def user_authenticate(self, name, password) -> Optional[bool]:
def user_authenticate(self, name: str, password: str) -> Optional[str]:
"""
使用Emby用户辅助完成用户认证
:param name: 用户名
:param password: 密码
:return: bool
:return: token or None
"""
# Emby认证
return self.emby.authenticate(name, password)
def webhook_parser(self, body: Any, form: Any, args: Any) -> Optional[dict]:

22
app/modules/emby/emby.py
View File

@ -6,10 +6,10 @@ from typing import List, Optional, Union, Dict
from app.core.config import settings
from app.log import logger
from app.schemas import RefreshMediaItem
from app.schemas.types import MediaType
from app.utils.http import RequestUtils
from app.utils.singleton import Singleton
from app.utils.string import StringUtils
from app.schemas.types import MediaType
class Emby(metaclass=Singleton):
@ -87,27 +87,37 @@ class Emby(metaclass=Singleton):
logger.error(f"连接Users出错：" + str(e))
return None
def authenticate(self, username: str, password: str) -> Optional[bool]:
def authenticate(self, username: str, password: str) -> Optional[str]:
"""
用户认证
:param username: 用户名
:param password: 密码
:return: 认证token
"""
if not self._host or not self._apikey:
return None
req_url = "%semby/Users/AuthenticateByName" % self._host
try:
res = RequestUtils(content_type="application/json").post_res(
res = RequestUtils(headers={
'X-Emby-Authorization': f'MediaBrowser Client="MoviePilot", '
f'Device="Axios", '
f'DeviceId="1", '
f'Version="10.8.0", '
f'Token="{self._apikey}"',
'Content-Type': 'application/json',
"Accept": "application/json"
}).post_res(
url=req_url,
data=json.dumps({
"Username": username,
"Pw": password,
"KeepMeLoggedIn": False
"Pw": password
})
)
if res:
auth_token = res.json().get("AccessToken")
if auth_token:
logger.info(f"用户 {username} Emby认证成功")
return True
return auth_token
else:
logger.error(f"Users/AuthenticateByName 未获取到返回数据")
except Exception as e:

5
app/modules/jellyfin/__init__.py
View File

@ -22,13 +22,14 @@ class JellyfinModule(_ModuleBase):
def init_setting(self) -> Tuple[str, Union[str, bool]]:
return "MEDIASERVER", "jellyfin"
def user_authenticate(self, name, password) -> Optional[bool]:
def user_authenticate(self, name: str, password: str) -> Optional[str]:
"""
使用Emby用户辅助完成用户认证
:param name: 用户名
:param password: 密码
:return: bool
:return: Token or None
"""
# Jellyfin认证
return self.jellyfin.authenticate(name, password)
def webhook_parser(self, body: Any, form: Any, args: Any) -> Optional[dict]:

17
app/modules/jellyfin/jellyfin.py
View File

@ -84,15 +84,26 @@ class Jellyfin(metaclass=Singleton):
logger.error(f"连接Users出错：" + str(e))
return None
def authenticate(self, username: str, password: str) -> Optional[bool]:
def authenticate(self, username: str, password: str) -> Optional[str]:
"""
用户认证
:param username: 用户名
:param password: 密码
:return: 认证成功返回token，否则返回None
"""
if not self._host or not self._apikey:
return None
req_url = "%sUsers/authenticatebyname" % self._host
try:
res = RequestUtils(content_type="application/json").post_res(
res = RequestUtils(headers={
'X-Emby-Authorization': f'MediaBrowser Client="MoviePilot", '
f'Device="Axios", '
f'DeviceId="1", '
f'Version="10.8.0", '
f'Token="{self._apikey}"',
'Content-Type': 'application/json',
"Accept": "application/json"
}).post_res(
url=req_url,
data=json.dumps({
"Username": username,
@ -103,7 +114,7 @@ class Jellyfin(metaclass=Singleton):
auth_token = res.json().get("AccessToken")
if auth_token:
logger.info(f"用户 {username} Jellyfin认证成功")
return True
return auth_token
else:
logger.error(f"Users/AuthenticateByName 未获取到返回数据")
except Exception as e: