def google_login(self, user):
db = DB()
GOOGLE_AUTH_URL = 'https://oauth2.googleapis.com/tokeninfo?id_token='
CORRECT_ISS_LIST = [
'accounts.google.com', 'https://accounts.google.com'
]
google_id = user['id']
google_token = user['token']
token_data = json.loads(
requests.get(GOOGLE_AUTH_URL + google_token).text)
if (token_data['iss']
not in CORRECT_ISS_LIST) or (token_data['sub'] != google_id):
return {'message': 'MODIFIED_TOKEN'}, 401
user_id = self.user_dao.google_login(user, db)
if user_id is None:
user_id = self.user_dao.get_user(google_id, db)
token = jwt.encode({
'id': user_id
},
SECRET_KEY['secret'],
algorithm=ALGORITHM['algorithm']).decode('utf-8')
if db:
db.close()
return {'token': token}, 200
def recommand_video_service(self):
db = DB()
try:
videos = []
for position in yaml['position_id']:
videos.append(
self.video_dao.recommand_video_by_us(position, db))
return videos
except pymysql.err.Error:
return {"message": "DATABASE ERROR"}, 500
finally:
if db:
db.close()
def get_video_detail(self, video_id):
db = DB()
try:
if video_id > yaml["video_max_id"] or video_id < yaml[
"video_min_id"]:
return yaml["no_response"]
video_detail, video_playlist = (
self.video_dao.get_video_detail(video_id, db),
self.video_dao.get_video_playlist(video_id, db),
)
return video_detail, video_playlist
except pymysql.err.Error:
return {"message": "DATABASE ERROR"}, 500
finally:
if db:
db.close()
def get_category_lists(self):
db = DB()
try:
response = {}
dict_stacks = {
'general_stacks': [],
'frontend_stacks': [],
'backend_stacks': []
}
contents_types_channels = self.video_dao.get_contents_types_channels(
db)
stacks = self.video_dao.get_stacks(db)
response['content_types'] = contents_types_channels[:4]
response['stacks'] = []
response['channels'] = contents_types_channels[4:]
for stack in stacks:
if stack['position_id'] == yaml['general']:
dict_stacks['general_stacks'].append(stack)
elif stack['position_id'] == yaml['frontend']:
dict_stacks['frontend_stacks'].append(stack)
else:
dict_stacks['backend_stacks'].append(stack)
for stack, position in dict_stacks.items():
response['stacks'].append({stack: position})
return response
except pymysql.err.Error:
return {"message": "DATABASE ERROR"}
finally:
if db:
db.close()
def get_video_lists(self, params):
db = DB()
try:
videos = []
contents_types = params["contents_types_id"]
stacks = params["stack_id"]
channels = params["channels_id"]
# 콘텐츠 타입 필터
if contents_types:
videos.append(
self.video_dao.get_contents_types_videos(
contents_types, db))
# 스택 필터
if contents_types is None and channels is None and stacks:
for stack in stacks:
stack_video = self.video_dao.get_stack_videos(stack, db)
# 검색된 배열이 없는 경우
if stack_video == 0:
videos.append([])
else:
videos.append(stack_video)
# 채널 필터
if contents_types is None and stacks is None and channels:
for channel in channels:
channel_video = self.video_dao.get_channel_videos(
channel, db)
# 검색된 배열이 없는 경우
if channel_video == 0:
videos.append([])
else:
videos.append(channel_video)
# 채널, 스택 필터
if contents_types is None and stacks and channels:
channel_videos = []
for channel in channels:
videos = []
for stack in stacks:
filters = {}
filters["stack_id"] = stack
filters["channel_id"] = channel
stack_videos = self.video_dao.get_videos(filters, db)
if stack_videos == 0:
stack_videos = ""
videos += stack_videos
channel_videos.append(videos)
return channel_videos
return videos
except pymysql.err.Error:
return {"message": "DATABASE ERROR"}, 500
finally:
if db:
db.close()