1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
import asyncio
import contextlib
import hashlib
import os
import shutil
import typing
import subprocess
@contextlib.contextmanager
def temp_dir(path: str):
os.mkdir(path)
try:
yield path
finally:
shutil.rmtree(path)
class TexGenerationError(Exception):
pass
class TexGenerator:
def __init__(self, storage_path: str, temp_path: str, task_timeout):
self._storage_path = os.path.abspath(storage_path)
self._temp_path = os.path.abspath(temp_path)
self._task_timeout = task_timeout
for p in (self._temp_path, self._storage_path):
try:
os.mkdir(p)
except FileExistsError:
pass
async def xelatex(self, tex_source: str) -> str:
tex_hash = hashlib.sha256(tex_source.encode('utf-8')).hexdigest()
cache_file_path = os.path.join(self._storage_path, f'{tex_hash}.pdf')
if os.path.exists(cache_file_path):
return cache_file_path
with temp_dir(os.path.join(self._temp_path, os.urandom(24).hex())) as workdir:
job_name = 'texput'
tex_file_path = os.path.join(self._temp_path, f'{job_name}.tex')
with open(tex_file_path, 'w', encoding='utf-8') as f:
f.write(tex_source)
with subprocess.Popen(
[
'xelatex',
'-interaction=nonstopmode',
'-halt-on-error',
tex_file_path,
],
cwd=workdir,
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT,
text=True,
shell=True,
) as proc:
stdout, stderr = proc.communicate(input='', timeout=self._task_timeout)
print('STDOUT', stdout)
print('STDERR', stderr)
if proc.returncode != 0:
raise TexGenerationError(f'xelatex process exited with non-zero code {proc.returncode}')
os.rename(os.path.join(workdir, f'{job_name}.pdf'), cache_file_path)
return cache_file_path
|
