pdf-auto-outline 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

pdf_auto_outline/__init__.py

@@ -1 +1 @@
-__version__ = '0.1.5'
+__version__ = '0.1.6'

pdf_auto_outline/main.py

@@ -1,10 +1,10 @@
 import pymupdf.layout
-from pymupdf import Point
 from time import perf_counter
 from multiprocessing import Pool
 import os
 import subprocess
 import argparse
+import tempfile
 
 SIOYEK = None
 
@@ -148,7 +148,12 @@ def align_toc_lvls(toc_entries: list) -> list:
 
     return toc_entries
 
-def generate_txtfile(toc_entries, txtfile='outline.txt') -> str:
+def get_tmpfile():
+    return tempfile.NamedTemporaryFile(
+            mode='w+', encoding='utf-8', delete=False, suffix='.txt'
+            )
+
+def generate_txtfile(toc_entries, txtfile=get_tmpfile()):
     import textwrap
     txt = textwrap.dedent("""\
     ============================================================
@@ -168,34 +173,37 @@ def generate_txtfile(toc_entries, txtfile='outline.txt') -> str:
         txt += '\n'.join(f"{' '*4 * (i[0] - 1)}{i[1]}  |  {i[2]}"
                         for i in toc_entries)
 
-    with open(txtfile, 'w', encoding='utf-8') as f:
-        f.write(txt)
+    txtfile.write(txt)
+    txtfile.flush()
+    txtfile.seek(0)
 
     return txtfile
 
 
-def parse_txtfile(txtfile='outline.txt', tablevel=2) -> list:
+
+def parse_txtfile(f, tablevel=2) -> list:
     toc_entries = []
-    with open(txtfile) as f:
-        if (c := f.read(1)) == 'C':
-            log('Outline not written')
-            exit()
-        elif c == '=':
-            lines = f.readlines()[7:]
-        else: lines = f.read()
-
-        for i in lines:
-            i = i.replace('\t', '    '*tablevel)
-            lvl = (len(i) - len(i.lstrip())) // 4 + 1
-            a = i.lstrip().split('  |  ')
-            if len(a) < 3:
-                toc_entries.append(
-                        [lvl, a[0], int(a[1])] 
-                )
-            else:
-                toc_entries.append(
-                        [lvl, a[0], int(a[1]), eval(a[2])]
-                )
+    if (c := f.read(1)) == 'C':
+        log('Outline not written')
+        exit()
+    elif c == '=':
+        lines = f.readlines()[7:]
+    else: lines = f.read()
+
+    for i in lines:
+        i = i.replace('\t', '    '*tablevel)
+        lvl = (len(i) - len(i.lstrip())) // 4 + 1
+        a = i.lstrip().split('  |  ')
+        if len(a) < 3:
+            toc_entries.append(
+                    [lvl, a[0], int(a[1])] 
+            )
+        else:
+            toc_entries.append(
+                    [lvl, a[0], int(a[1]), eval(a[2])]
+            )
+    
+    f.close()
 
     return toc_entries
 
@@ -213,15 +221,12 @@ def get_toc_custom(doc) -> list:
     toc_entries = [[*i[:3], i[3].get('to')[1]] for i in doc.get_toc(False)]
     return toc_entries
 
-def edit_txtfile(txtfile='outline.txt'):
-    # editor = os.environ.get('EDITOR', 'notepad' if os.name == 'nt' else 'vi')
-    # editor = os.environ.get('EDITOR', 'start' if os.name == 'nt' else 'xdg-open')
-    name = os.name
-    if name == 'nt':
-        subprocess.run(['start', '/WAIT', txtfile], shell=True)
+def edit_txtfile(f):
+    if os.name == 'nt':
+        subprocess.run(['start', '/WAIT', f.name], shell=True)
     else: # name == 'posix':
         editor = os.environ.get('EDITOR', 'vi')  
-        subprocess.run([editor, txtfile])
+        subprocess.run([editor, f.name])
 
 def main():
     parser = argparse.ArgumentParser(prog='pdfao')
@@ -234,19 +239,19 @@ def main():
     parser.add_argument('-i', '--infile', type=str, metavar='<file>', help='write toc from file to pdf')
     parser.add_argument('-t', '--tablevel', type=int, metavar='<n>', help='tab = n toc nesting levels (default 2)', default=2)
     parser.add_argument('--sioyek', type=str, metavar='<path>', help='for users of the Sioyek pdf viewer')
-    parser.add_argument('--version', action='version', version='%(prog)s 0.1.5')
+    parser.add_argument('--version', action='version', version='%(prog)s 0.1.6')
 
     args = parser.parse_args()
 
+    if args.out:
+        args.out = os.path.join(
+                os.path.dirname(args.filename),
+                args.out)
+
     if args.sioyek:
         from sioyek.sioyek import Sioyek
         global SIOYEK
         SIOYEK = Sioyek(args.sioyek)
-        if args.out:
-            args.out = os.path.join(
-                    os.path.dirname(args.filename),
-                    args.out
-                )
         # local_db = args.sioyek[1]
         # shared_db = args.sioyek[2]
         # pdf_path = args.sioyek[3]
@@ -255,14 +260,15 @@ def main():
     if args.edit or args.superedit:
         doc = pymupdf.Document(args.filename)
         if args.superedit:
-            generate_txtfile(doc.get_toc(False))
+            f = generate_txtfile(doc.get_toc(False))
         else:
-            generate_txtfile(get_toc_custom(doc))
-        edit_txtfile()
-        toc_entries = parse_txtfile(tablevel=args.tablevel)
+            f = generate_txtfile(get_toc_custom(doc))
+        edit_txtfile(f)
+        toc_entries = parse_txtfile(f, args.tablevel)
         embed_toc(args.filename, toc_entries, args.out)
+        os.remove(f.name)
     elif args.infile:
-        toc_entries = parse_txtfile(args.infile, args.tablevel)
+        toc_entries = parse_txtfile(open(args.infile, encoding='utf-8'), args.tablevel)
         embed_toc(args.filename, toc_entries, args.out)
     else: # generate toc
         start = perf_counter()
@@ -274,10 +280,11 @@ def main():
         if args.straight:
             embed_toc(args.filename, toc_entries, args.out)
         else:
-            generate_txtfile(toc_entries)
-            edit_txtfile()
-            toc_entries = parse_txtfile(tablevel=args.tablevel)
+            f = generate_txtfile(toc_entries)
+            edit_txtfile(f)
+            toc_entries = parse_txtfile(f, args.tablevel)
             embed_toc(args.filename, toc_entries, args.out)
+            os.remove(f.name)
 
     # if args.sioyek and not args.out:
     #     to_hash = get_md5_hash(args.filename)

pdf_auto_outline/test.py

@@ -0,0 +1,4 @@
+txt = "aa\nbb\ncc"
+
+for i in txt:
+    print(i)

{pdf_auto_outline-0.1.5.dist-info → pdf_auto_outline-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pdf-auto-outline
-Version: 0.1.5
+Version: 0.1.6
 Summary: Automatically generate and edit PDF table of contents / outline
 Author: Rossikos
 Author-email: Rossikos <216631970+rossikos@users.noreply.github.com>
@@ -82,7 +82,6 @@ The optional part can be one of:
 ```
   |  None                  	same as not including it   
   |  241.2					y-ordinate 
-  |  Point(72.0, 363.9)		x and y coords
   |  {<dictionary>}			dictionary with more attributes for the ToC entry
 ```
 
@@ -92,8 +91,10 @@ Example commands; add to `prefs_user.config`.
 
 ```
 new_command _gen_toc pdfao "%{file_path}" --sioyek path/to/sioyek -mp 4
-new_command _edit_toc pdfao "%{file_path}" --sioyek path/to/sioyek -e
+new_command _edit_toc pdfao "%{file_path}"
 ```
 
+The sioyek library and flag are optional; they allow logging to the status bar. This is more useful for ToC generation where you may want a progress bar. 
+
 If you don't wish to install from PyPI, download source and use `python3 -m path/to/src/pdf_auto_outline` in place of `pdfao`.
 

pdf_auto_outline-0.1.6.dist-info/RECORD

@@ -0,0 +1,8 @@
+pdf_auto_outline/__init__.py,sha256=gW5NUxwGdPsiQjn0cOuuQT11pfthByI5DITDg_HMhLQ,22
+pdf_auto_outline/__main__.py,sha256=mRKsAFeG5R17vTYubIKregAve4vnKc-nk7jY3tcK4wI,78
+pdf_auto_outline/main.py,sha256=AWbXZNVw5e34v005qtoCGyTzNbT3T1mOo6xeLl-Q82k,9883
+pdf_auto_outline/test.py,sha256=2wIjM8xqnDjHiokFF0WUNw0o_oTGW91V-MifHoLw4Rw,47
+pdf_auto_outline-0.1.6.dist-info/WHEEL,sha256=5w2T7AS2mz1-rW9CNagNYWRCaB0iQqBMYLwKdlgiR4Q,78
+pdf_auto_outline-0.1.6.dist-info/entry_points.txt,sha256=HBvhmxJs8hHqbbpJmVTbBH3xy19Hk655O_ySwFC_53w,100
+pdf_auto_outline-0.1.6.dist-info/METADATA,sha256=Rw9x1kQxghKSGVu31qbnLASX87QMjAeUzg2JMAecupU,3287
+pdf_auto_outline-0.1.6.dist-info/RECORD,,

pdf_auto_outline-0.1.5.dist-info/RECORD

@@ -1,7 +0,0 @@
-pdf_auto_outline/__init__.py,sha256=DQiXHqQ3C29f_--ye7aVMDPnITrl9M0zMU42LYY48zE,22
-pdf_auto_outline/__main__.py,sha256=mRKsAFeG5R17vTYubIKregAve4vnKc-nk7jY3tcK4wI,78
-pdf_auto_outline/main.py,sha256=d5D6qlnWKU1_bGiUKc82mLDwxhH6D-0WRBoxiDHzHsU,10020
-pdf_auto_outline-0.1.5.dist-info/WHEEL,sha256=5w2T7AS2mz1-rW9CNagNYWRCaB0iQqBMYLwKdlgiR4Q,78
-pdf_auto_outline-0.1.5.dist-info/entry_points.txt,sha256=HBvhmxJs8hHqbbpJmVTbBH3xy19Hk655O_ySwFC_53w,100
-pdf_auto_outline-0.1.5.dist-info/METADATA,sha256=cAkatx-gjHs3-Sg9cQNeCQosrxHikoPjo1Q2AYbHVHw,3198
-pdf_auto_outline-0.1.5.dist-info/RECORD,,