summaryrefslogtreecommitdiff
path: root/pagesindexgen.py
blob: acb3aaa203fb2af0c2caaf6c51ab0f18c89ed2fe (plain) (blame)
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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#! /usr/bin/env python
# -*- coding: utf8 -*-
#
# Copyright (C) 2016  Andrei Karas (4144)

import os
import re
import sys

filt = re.compile(".+", re.IGNORECASE)
indexName = "index.htm"

def showHelp():
    print("PagesIndexGen v0.1")
    print("Usage:")
    print("  ./pagesindexgen.py filedir sitedir template")
    print("Example:")
    print("  ./pagesindexgen.py")
    print("  ./pagesindexgen.py .")
    print("  ./pagesindexgen.py . /")
    print("  ./pagesindexgen.py . / default")

def readTemplate(name):
    with open(template + os.path.sep + name, "r") as f:
        return f.read()

def saveTemplate(name, data):
    with open(name + os.path.sep + indexName, "w") as w:
        w.write(data)

def getFullRelativePath(relativeDir, fileName):
    if len(relativeDir) > 0:
        relativePath = relativeDir + os.path.sep + fileName
    else:
        relativePath = fileName
    return relativePath

def buildIndex(curDir, relativeDir, dirName, dirs, files, backLink):
    indexTemplate = readTemplate("index.tpl")
    dirTemplate = readTemplate("dir.tpl")
    fileTemplate = readTemplate("file.tpl")

    data = ""

    if backLink == True:
        backTemplate = readTemplate("back.tpl")
        data = data + backTemplate.format(
            name = "..",
            url = ".." + os.path.sep + indexName,
            sitedir = dirName,
            urldir = relativeDir,
            filedir = curDir,
        )

    for path in dirs:
        data = data + dirTemplate.format(
            name = path,
            url = path + os.path.sep + indexName,
            sitedir = dirName,
            urldir = relativeDir,
            filedir = curDir,
        )

    for name in files:
        fullPath = os.path.abspath(curDir + os.path.sep + name)
        relativePath = getFullRelativePath(relativeDir, name)
        data = data + fileTemplate.format(
            name = name,
            url = name,
            sitedir = dirName,
            urldir = relativeDir,
            filedir = curDir,
        )

    saveTemplate(curDir, indexTemplate.format(
        sitedir = dirName,
        url = relativeDir,
        filedir = curDir,
        body = data
    ))

def processDir(curDir, relativeDir, dirName, backLink):
    curFiles = os.listdir(curDir)
    dirs = []
    files = []
    for fileName in curFiles:
        if fileName[0] == "." or fileName == indexName:
            continue
        fullPath = os.path.abspath(curDir + os.path.sep + fileName)
        relativePath = getFullRelativePath(relativeDir, fileName)
        if not os.path.isfile(fullPath):
            dirs.append(fileName)
        elif filt.search(fileName):
            files.append(fileName)

    dirs.sort()
    files.sort()

    buildIndex(curDir, relativeDir, dirName, dirs, files, backLink)

    for path in dirs:
        fullPath = os.path.abspath(curDir + os.path.sep + path)
        relativePath = getFullRelativePath(relativeDir, path)
        processDir(fullPath, relativePath, path, True)

num = len(sys.argv)
if num < 1 or num > 4:
    showHelp()
    exit(1)
if num > 1:
    filePath = sys.argv[1]
else:
    filePath = "."
if num > 2:
    sitePath = sys.argv[2]
else:
    sitePath = ""
if num > 3:
    template = sys.argv[3]
else:
    template = "default"
template = "templates" + os.path.sep + template

processDir(filePath, sitePath, "", False)