papershelf/tools.py at master · daniel-e/papershelf · GitHub

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
66
67
68
69
70
71
import os, sys, re, sys
import tempfile, urllib2

# TODO settings
import settings

def child(pdffile):
  f = open("/dev/null", "w")
  os.dup2(sys.stdout.fileno(), f.fileno())
  os.dup2(sys.stderr.fileno(), f.fileno())
  s = settings.Settings()
  os.execvp(s.vars["pdfviewer"], [s.vars["pdfviewer"], pdffile])
  sys.exit(1)

def short_name(n):
  n = n.lower().replace(' ', '_')
  n = re.sub('-', '_', n)
  n = re.sub('\'', '', n)
  n = re.sub(':', '', n)
  n = re.sub('^._', '', n)
  n = re.sub('^.._', '', n)
  n = re.sub('^..._', '', n)
  n = re.sub('_._', '_', n)
  n = re.sub('_.._', '_', n)
  n = re.sub('_..._', '_', n)
  while len(n) > 30:
    p = n.rfind("_")
    n = n[:p]
  return n

def next_number():
  s = settings.Settings()
  path = s.vars["pdflocation"]
  max = 0
  files = [f for f in os.listdir(path) if os.path.isfile(path + "/" + f)]
  for f in files:
    c = re.compile('^([0-9]+)_.*')
    m = c.match(f)
    if m:
      if int(m.group(1)) > max: max = int(m.group(1))
  return max + 1

def download_pdf(url):
  f = tempfile.NamedTemporaryFile(delete = False, suffix = ".getpdf.pdf")
  pdf = urllib2.urlopen(url).read()
  f.write(pdf)
  f.close()
  return f.name

def extern_pdf_view(fname):
  pid = os.fork()
  if pid == 0:
    child(fname)
    sys.exit(1)

def wrap_it(s, arr, res):
  s = s.strip()
  if len(arr) == 0:
    if len(s) > 0:
      res.append(s)
    return res
  if len(s) + len(arr[0]) > 30 and len(s) > 0:
    res.append(s)
    return wrap_it("", arr, res)
  else:
    return wrap_it(s + " " + arr[0], arr[1:], res)

def wrap_string(s):
  r = []
  ss = [i.strip() for i in s.split(" ") if len(s.strip()) > 0]
  return wrap_it("", ss, r)