mirror of
https://github.com/janeczku/calibre-web
synced 2024-11-17 15:24:53 +00:00
Fix for #890
This commit is contained in:
parent
b9c3a3fcea
commit
9f64a96502
100
cps/worker.py
100
cps/worker.py
@ -195,15 +195,18 @@ class WorkerThread(threading.Thread):
|
|||||||
doLock = threading.Lock()
|
doLock = threading.Lock()
|
||||||
doLock.acquire()
|
doLock.acquire()
|
||||||
if self.current != self.last:
|
if self.current != self.last:
|
||||||
|
index = self.current
|
||||||
doLock.release()
|
doLock.release()
|
||||||
if self.queue[self.current]['taskType'] == TASK_EMAIL:
|
if self.queue[index]['taskType'] == TASK_EMAIL:
|
||||||
self._send_raw_email()
|
self._send_raw_email()
|
||||||
if self.queue[self.current]['taskType'] == TASK_CONVERT:
|
if self.queue[index]['taskType'] == TASK_CONVERT:
|
||||||
self._convert_any_format()
|
self._convert_any_format()
|
||||||
if self.queue[self.current]['taskType'] == TASK_CONVERT_ANY:
|
if self.queue[index]['taskType'] == TASK_CONVERT_ANY:
|
||||||
self._convert_any_format()
|
self._convert_any_format()
|
||||||
# TASK_UPLOAD is handled implicitly
|
# TASK_UPLOAD is handled implicitly
|
||||||
|
doLock.acquire()
|
||||||
self.current += 1
|
self.current += 1
|
||||||
|
doLock.release()
|
||||||
else:
|
else:
|
||||||
doLock.release()
|
doLock.release()
|
||||||
if main_thread.is_alive():
|
if main_thread.is_alive():
|
||||||
@ -226,6 +229,8 @@ class WorkerThread(threading.Thread):
|
|||||||
self.last = len(self.queue)
|
self.last = len(self.queue)
|
||||||
|
|
||||||
def get_taskstatus(self):
|
def get_taskstatus(self):
|
||||||
|
doLock = threading.Lock()
|
||||||
|
doLock.acquire()
|
||||||
if self.current < len(self.queue):
|
if self.current < len(self.queue):
|
||||||
if self.UIqueue[self.current]['stat'] == STAT_STARTED:
|
if self.UIqueue[self.current]['stat'] == STAT_STARTED:
|
||||||
if self.queue[self.current]['taskType'] == TASK_EMAIL:
|
if self.queue[self.current]['taskType'] == TASK_EMAIL:
|
||||||
@ -234,30 +239,39 @@ class WorkerThread(threading.Thread):
|
|||||||
self.UIqueue[self.current]['rt'] = self.UIqueue[self.current]['formRuntime'].days*24*60 \
|
self.UIqueue[self.current]['rt'] = self.UIqueue[self.current]['formRuntime'].days*24*60 \
|
||||||
+ self.UIqueue[self.current]['formRuntime'].seconds \
|
+ self.UIqueue[self.current]['formRuntime'].seconds \
|
||||||
+ self.UIqueue[self.current]['formRuntime'].microseconds
|
+ self.UIqueue[self.current]['formRuntime'].microseconds
|
||||||
|
doLock.release()
|
||||||
return self.UIqueue
|
return self.UIqueue
|
||||||
|
|
||||||
def _convert_any_format(self):
|
def _convert_any_format(self):
|
||||||
# convert book, and upload in case of google drive
|
# convert book, and upload in case of google drive
|
||||||
self.UIqueue[self.current]['stat'] = STAT_STARTED
|
doLock = threading.Lock()
|
||||||
self.queue[self.current]['starttime'] = datetime.now()
|
doLock.acquire()
|
||||||
self.UIqueue[self.current]['formStarttime'] = self.queue[self.current]['starttime']
|
index = self.current
|
||||||
curr_task = self.queue[self.current]['taskType']
|
doLock.release()
|
||||||
|
self.UIqueue[index]['stat'] = STAT_STARTED
|
||||||
|
self.queue[index]['starttime'] = datetime.now()
|
||||||
|
self.UIqueue[index]['formStarttime'] = self.queue[self.current]['starttime']
|
||||||
|
curr_task = self.queue[index]['taskType']
|
||||||
filename = self._convert_ebook_format()
|
filename = self._convert_ebook_format()
|
||||||
if filename:
|
if filename:
|
||||||
if config.config_use_google_drive:
|
if config.config_use_google_drive:
|
||||||
gdriveutils.updateGdriveCalibreFromLocal()
|
gdriveutils.updateGdriveCalibreFromLocal()
|
||||||
if curr_task == TASK_CONVERT:
|
if curr_task == TASK_CONVERT:
|
||||||
self.add_email(self.queue[self.current]['settings']['subject'], self.queue[self.current]['path'],
|
self.add_email(self.queue[index]['settings']['subject'], self.queue[index]['path'],
|
||||||
filename, self.queue[self.current]['settings'], self.queue[self.current]['kindle'],
|
filename, self.queue[index]['settings'], self.queue[index]['kindle'],
|
||||||
self.UIqueue[self.current]['user'], self.queue[self.current]['title'],
|
self.UIqueue[index]['user'], self.queue[index]['title'],
|
||||||
self.queue[self.current]['settings']['body'])
|
self.queue[index]['settings']['body'])
|
||||||
|
|
||||||
def _convert_ebook_format(self):
|
def _convert_ebook_format(self):
|
||||||
error_message = None
|
error_message = None
|
||||||
file_path = self.queue[self.current]['file_path']
|
doLock = threading.Lock()
|
||||||
bookid = self.queue[self.current]['bookid']
|
doLock.acquire()
|
||||||
format_old_ext = u'.' + self.queue[self.current]['settings']['old_book_format'].lower()
|
index = self.current
|
||||||
format_new_ext = u'.' + self.queue[self.current]['settings']['new_book_format'].lower()
|
doLock.release()
|
||||||
|
file_path = self.queue[index]['file_path']
|
||||||
|
bookid = self.queue[index]['bookid']
|
||||||
|
format_old_ext = u'.' + self.queue[index]['settings']['old_book_format'].lower()
|
||||||
|
format_new_ext = u'.' + self.queue[index]['settings']['new_book_format'].lower()
|
||||||
|
|
||||||
# check to see if destination format already exists -
|
# check to see if destination format already exists -
|
||||||
# if it does - mark the conversion task as complete and return a success
|
# if it does - mark the conversion task as complete and return a success
|
||||||
@ -265,8 +279,8 @@ class WorkerThread(threading.Thread):
|
|||||||
if os.path.isfile(file_path + format_new_ext):
|
if os.path.isfile(file_path + format_new_ext):
|
||||||
log.info("Book id %d already converted to %s", bookid, format_new_ext)
|
log.info("Book id %d already converted to %s", bookid, format_new_ext)
|
||||||
cur_book = db.session.query(db.Books).filter(db.Books.id == bookid).first()
|
cur_book = db.session.query(db.Books).filter(db.Books.id == bookid).first()
|
||||||
self.queue[self.current]['path'] = file_path
|
self.queue[index]['path'] = file_path
|
||||||
self.queue[self.current]['title'] = cur_book.title
|
self.queue[index]['title'] = cur_book.title
|
||||||
self._handleSuccess()
|
self._handleSuccess()
|
||||||
return file_path + format_new_ext
|
return file_path + format_new_ext
|
||||||
else:
|
else:
|
||||||
@ -304,13 +318,13 @@ class WorkerThread(threading.Thread):
|
|||||||
else:'''
|
else:'''
|
||||||
command = [config.config_converterpath, (file_path + format_old_ext),
|
command = [config.config_converterpath, (file_path + format_old_ext),
|
||||||
(file_path + format_new_ext)]
|
(file_path + format_new_ext)]
|
||||||
index = 3
|
quotes_index = 3
|
||||||
if config.config_calibre:
|
if config.config_calibre:
|
||||||
parameters = config.config_calibre.split(" ")
|
parameters = config.config_calibre.split(" ")
|
||||||
for param in parameters:
|
for param in parameters:
|
||||||
command.append(param)
|
command.append(param)
|
||||||
quotes.append(index)
|
quotes.append(quotes_index)
|
||||||
index += 1
|
quotes_index += 1
|
||||||
p = process_open(command, quotes)
|
p = process_open(command, quotes)
|
||||||
# p = subprocess.Popen(command, stdout=subprocess.PIPE, universal_newlines=True)
|
# p = subprocess.Popen(command, stdout=subprocess.PIPE, universal_newlines=True)
|
||||||
except OSError as e:
|
except OSError as e:
|
||||||
@ -338,7 +352,7 @@ class WorkerThread(threading.Thread):
|
|||||||
# parse progress string from calibre-converter
|
# parse progress string from calibre-converter
|
||||||
progress = re.search(r"(\d+)%\s.*", nextline)
|
progress = re.search(r"(\d+)%\s.*", nextline)
|
||||||
if progress:
|
if progress:
|
||||||
self.UIqueue[self.current]['progress'] = progress.group(1) + ' %'
|
self.UIqueue[index]['progress'] = progress.group(1) + ' %'
|
||||||
|
|
||||||
# process returncode
|
# process returncode
|
||||||
check = p.returncode
|
check = p.returncode
|
||||||
@ -359,12 +373,12 @@ class WorkerThread(threading.Thread):
|
|||||||
cur_book = db.session.query(db.Books).filter(db.Books.id == bookid).first()
|
cur_book = db.session.query(db.Books).filter(db.Books.id == bookid).first()
|
||||||
if os.path.isfile(file_path + format_new_ext):
|
if os.path.isfile(file_path + format_new_ext):
|
||||||
new_format = db.Data(name=cur_book.data[0].name,
|
new_format = db.Data(name=cur_book.data[0].name,
|
||||||
book_format=self.queue[self.current]['settings']['new_book_format'].upper(),
|
book_format=self.queue[index]['settings']['new_book_format'].upper(),
|
||||||
book=bookid, uncompressed_size=os.path.getsize(file_path + format_new_ext))
|
book=bookid, uncompressed_size=os.path.getsize(file_path + format_new_ext))
|
||||||
cur_book.data.append(new_format)
|
cur_book.data.append(new_format)
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
self.queue[self.current]['path'] = cur_book.path
|
self.queue[index]['path'] = cur_book.path
|
||||||
self.queue[self.current]['title'] = cur_book.title
|
self.queue[index]['title'] = cur_book.title
|
||||||
if config.config_use_google_drive:
|
if config.config_use_google_drive:
|
||||||
os.remove(file_path + format_old_ext)
|
os.remove(file_path + format_old_ext)
|
||||||
self._handleSuccess()
|
self._handleSuccess()
|
||||||
@ -430,16 +444,20 @@ class WorkerThread(threading.Thread):
|
|||||||
|
|
||||||
|
|
||||||
def _send_raw_email(self):
|
def _send_raw_email(self):
|
||||||
self.queue[self.current]['starttime'] = datetime.now()
|
doLock = threading.Lock()
|
||||||
self.UIqueue[self.current]['formStarttime'] = self.queue[self.current]['starttime']
|
doLock.acquire()
|
||||||
self.UIqueue[self.current]['stat'] = STAT_STARTED
|
index = self.current
|
||||||
obj=self.queue[self.current]
|
doLock.release()
|
||||||
|
self.queue[index]['starttime'] = datetime.now()
|
||||||
|
self.UIqueue[index]['formStarttime'] = self.queue[index]['starttime']
|
||||||
|
self.UIqueue[index]['stat'] = STAT_STARTED
|
||||||
|
obj=self.queue[index]
|
||||||
# create MIME message
|
# create MIME message
|
||||||
msg = MIMEMultipart()
|
msg = MIMEMultipart()
|
||||||
msg['Subject'] = self.queue[self.current]['subject']
|
msg['Subject'] = self.queue[index]['subject']
|
||||||
msg['Message-Id'] = make_msgid('calibre-web')
|
msg['Message-Id'] = make_msgid('calibre-web')
|
||||||
msg['Date'] = formatdate(localtime=True)
|
msg['Date'] = formatdate(localtime=True)
|
||||||
text = self.queue[self.current]['text']
|
text = self.queue[index]['text']
|
||||||
msg.attach(MIMEText(text.encode('UTF-8'), 'plain', 'UTF-8'))
|
msg.attach(MIMEText(text.encode('UTF-8'), 'plain', 'UTF-8'))
|
||||||
if obj['attachment']:
|
if obj['attachment']:
|
||||||
result = get_attachment(obj['filepath'], obj['attachment'])
|
result = get_attachment(obj['filepath'], obj['attachment'])
|
||||||
@ -506,15 +524,23 @@ class WorkerThread(threading.Thread):
|
|||||||
|
|
||||||
def _handleError(self, error_message):
|
def _handleError(self, error_message):
|
||||||
log.error(error_message)
|
log.error(error_message)
|
||||||
self.UIqueue[self.current]['stat'] = STAT_FAIL
|
doLock = threading.Lock()
|
||||||
self.UIqueue[self.current]['progress'] = "100 %"
|
doLock.acquire()
|
||||||
self.UIqueue[self.current]['formRuntime'] = datetime.now() - self.queue[self.current]['starttime']
|
index = self.current
|
||||||
self.UIqueue[self.current]['message'] = error_message
|
doLock.release()
|
||||||
|
self.UIqueue[index]['stat'] = STAT_FAIL
|
||||||
|
self.UIqueue[index]['progress'] = "100 %"
|
||||||
|
self.UIqueue[index]['formRuntime'] = datetime.now() - self.queue[self.current]['starttime']
|
||||||
|
self.UIqueue[index]['message'] = error_message
|
||||||
|
|
||||||
def _handleSuccess(self):
|
def _handleSuccess(self):
|
||||||
self.UIqueue[self.current]['stat'] = STAT_FINISH_SUCCESS
|
doLock = threading.Lock()
|
||||||
self.UIqueue[self.current]['progress'] = "100 %"
|
doLock.acquire()
|
||||||
self.UIqueue[self.current]['formRuntime'] = datetime.now() - self.queue[self.current]['starttime']
|
index = self.current
|
||||||
|
doLock.release()
|
||||||
|
self.UIqueue[index]['stat'] = STAT_FINISH_SUCCESS
|
||||||
|
self.UIqueue[index]['progress'] = "100 %"
|
||||||
|
self.UIqueue[index]['formRuntime'] = datetime.now() - self.queue[self.current]['starttime']
|
||||||
|
|
||||||
|
|
||||||
_worker = WorkerThread()
|
_worker = WorkerThread()
|
||||||
|
Loading…
Reference in New Issue
Block a user