Shiloh/brotherton-erpnext
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: Separate function to get bulk and single youtube stats

Browse Source
This commit is contained in:
marination 2020-09-10 18:27:20 +05:30
parent 799d663398
commit 37b99d202d
1 changed files with 19 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
erpnext/utilities/doctype/video/video.py
View File

@ -14,31 +14,21 @@ from pyyoutube import Api
class Video(Document): class Video(Document):
def validate(self): def validate(self):
self.set_video_id() if self.provider == "YouTube" and is_tracking_enabled():
self.set_youtube_statistics() self.set_video_id()
self.set_youtube_statistics()
def set_video_id(self): def set_video_id(self):
if self.provider == "YouTube" and self.url and not self.get("youtube_video_id"): if self.url and not self.get("youtube_video_id"):
self.youtube_video_id = get_id_from_url(self.url) self.youtube_video_id = get_id_from_url(self.url)
def set_youtube_statistics(self, video_ids=None, update=True): def set_youtube_statistics(self):
if self.provider == "YouTube" and not is_tracking_enabled():
return
api_key = frappe.db.get_single_value("Video Settings", "api_key") api_key = frappe.db.get_single_value("Video Settings", "api_key")
api = Api(api_key=api_key) api = Api(api_key=api_key)
try: try:
video_id = video_ids or self.youtube_video_id video = api.get_video_by_id(video_id=self.youtube_video_id)
video = api.get_video_by_id(video_id=video_id) video_stats = video.items[0].to_dict().get('statistics')
if video_ids:
video_stats = video.items
else:
video_stats = video.items[0].to_dict().get('statistics')
if not update:
return video_stats
self.like_count = video_stats.get('likeCount') self.like_count = video_stats.get('likeCount')
self.view_count = video_stats.get('viewCount') self.view_count = video_stats.get('viewCount')
@ -49,7 +39,6 @@ class Video(Document):
title = "Failed to Update YouTube Statistics for Video: {0}".format(self.name) title = "Failed to Update YouTube Statistics for Video: {0}".format(self.name)
frappe.log_error(title + "\n\n" + frappe.get_traceback(), title=title) frappe.log_error(title + "\n\n" + frappe.get_traceback(), title=title)
def is_tracking_enabled(): def is_tracking_enabled():
return frappe.db.get_single_value("Video Settings", "enable_youtube_tracking") return frappe.db.get_single_value("Video Settings", "enable_youtube_tracking")
@ -107,10 +96,20 @@ def get_id_from_url(url):
@frappe.whitelist() @frappe.whitelist()
def batch_update_youtube_data(): def batch_update_youtube_data():
def get_youtube_statistics(video_ids):
api_key = frappe.db.get_single_value("Video Settings", "api_key")
api = Api(api_key=api_key)
try:
video = api.get_video_by_id(video_id=video_ids)
video_stats = video.items
return video_stats
except Exception:
title = "Failed to Update YouTube Statistics"
frappe.log_error(title + "\n\n" + frappe.get_traceback(), title=title)
def prepare_and_set_data(video_list): def prepare_and_set_data(video_list):
video_ids = get_formatted_ids(video_list) video_ids = get_formatted_ids(video_list)
video_doc = frappe.new_doc("Video") stats = get_youtube_statistics(video_ids)
stats = video_doc.set_youtube_statistics(video_ids=video_ids, update=False)
set_youtube_data(stats) set_youtube_data(stats)
def set_youtube_data(entries): def set_youtube_data(entries):