mirror of https://github.com/tLDP/python-tldp
119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
#! /usr/bin/python
|
|
|
|
from __future__ import print_function
|
|
|
|
import os
|
|
import sys
|
|
import logging
|
|
|
|
from .utils import logger
|
|
from .guess import knownextensions
|
|
|
|
# class SourceTree(object):
|
|
#
|
|
# def __init__(self, root, config):
|
|
# pass
|
|
|
|
|
|
class SourceDir(object):
|
|
|
|
def __repr__(self):
|
|
return self.dirname
|
|
|
|
def __init__(self, dirname):
|
|
self.dirname = os.path.abspath(dirname)
|
|
self.docs = list()
|
|
if not os.path.exists(dirname):
|
|
raise OSError("[Errno 2] No such file or directory: " + dirname)
|
|
logger.info("Time to go for an enumeration stroll in %s", dirname)
|
|
self.enumerateDocuments()
|
|
|
|
def enumerateDocuments(self):
|
|
for fname in os.listdir(self.dirname):
|
|
possible = os.path.join(self.dirname, fname)
|
|
if os.path.isfile(possible):
|
|
self.docs.append(SourceDocument(possible))
|
|
elif os.path.isdir(fname):
|
|
stem = os.path.basename(fname)
|
|
for ext in knownextensions:
|
|
possible = os.path.join(self.dirname, fname, stem + ext)
|
|
if os.path.isfile(possible):
|
|
self.docs.append(SourceDocument(possible))
|
|
logger.info("Discovered %s documents in %s",
|
|
len(self.docs), self.dirname)
|
|
|
|
|
|
class SourceDocument(object):
|
|
|
|
def __repr__(self):
|
|
return self.filename
|
|
|
|
def __init__(self, filename):
|
|
# -- canonicalize the pathname we are given.
|
|
self.filename = os.path.abspath(filename)
|
|
if not os.path.exists(self.filename):
|
|
raise OSError("Missing source document: " + self.filename)
|
|
|
|
self.dirname, self.basename = os.path.split(self.filename)
|
|
self.stem, self.ext = os.path.splitext(self.basename)
|
|
self.stat = os.stat(self.filename)
|
|
|
|
self.doctype = None
|
|
self.resources = False # -- assume no ./images/, ./resources/
|
|
self.singlefile = True # -- assume only one file
|
|
parentdir = os.path.basename(self.dirname)
|
|
if parentdir == self.stem:
|
|
self.singlefile = False
|
|
for rdir in ('resources', 'images'):
|
|
if os.path.exists(os.path.join(self.dirname, rdir)):
|
|
self.resources = True
|
|
|
|
def doctype(self):
|
|
if self.doctype is None:
|
|
self.doctype = guess(self.filename)
|
|
return self.doctype
|
|
|
|
|
|
class OutputDocument(object):
|
|
|
|
formats = {'pdf': '.pdf',
|
|
'txt': '.txt',
|
|
'html': '.html',
|
|
'htmls': '-single.html', }
|
|
|
|
def __init__(self, filename):
|
|
pass
|
|
|
|
@property
|
|
def txt(self):
|
|
return os.path.join(self.dirname, self.stem, '.txt')
|
|
|
|
|
|
class OutputDir(object):
|
|
|
|
def __init__(self, dirname):
|
|
self.dirname = os.path.abspath(dirname)
|
|
self.parent = os.path.dirname(dirname)
|
|
self.stem = os.path.basename(dirname)
|
|
self.members = list()
|
|
|
|
def mkdir(self):
|
|
if not os.path.exists(self.parent):
|
|
raise OSError("Missing parent directory: " + self.parent)
|
|
os.mkdir(self.dirname)
|
|
return os.path.exists(self.dirname)
|
|
|
|
@property
|
|
def members(self):
|
|
return os.path.exists(self.dirname)
|
|
|
|
@property
|
|
def exists(self):
|
|
return os.path.exists(self.dirname)
|
|
|
|
@property
|
|
def isComplete(self):
|
|
return all(self.pdf, self.html, self.htmls, self.txt)
|
|
|
|
# -- end of file
|