-
Notifications
You must be signed in to change notification settings - Fork 36
/
pyh.py
199 lines (164 loc) · 5.97 KB
/
pyh.py
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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
# coding:utf-8
# @file: pyh.py
# @purpose: a HTML tag generator
# @author: Emmanuel Turlay <turlay@cern.ch>
# TODO:追踪该标签所需要的缩紧层级。改善标签嵌套时输出的格式
__doc__ = """The pyh.py module is the core of the PyH package. PyH lets you
generate HTML tags from within your python code.
See http://code.google.com/p/pyh/ for documentation.
"""
__author__ = "Emmanuel Turlay <turlay@cern.ch>"
__version__ = '$Revision: 63 $'
__date__ = '$Date: 2010-05-21 03:09:03 +0200 (Fri, 21 May 2010) $'
from sys import _getframe, stdout, modules, version
nOpen = {}
nl = '\n'
doctype = '<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">\n'
charset = '<meta http-equiv="Content-Type" content="text/html;charset=utf-8" />\n'
# 添加style,colgroup标签,thead，取消了一个重复的script标签
tags = ['html', 'body', 'head', 'link', 'meta', 'div', 'p', 'form', 'legend',
'input', 'select', 'span', 'b', 'i', 'option', 'img', 'style',
'table', 'tr', 'td', 'th', 'h1', 'h2', 'h3', 'h4', 'h5', 'h6',
'fieldset', 'a', 'title', 'body', 'head', 'title', 'script', 'br', 'table',
'tbody','ul', 'li', 'ol', 'colgroup', 'thead']
selfClose = ['input', 'img', 'link', 'br']
class Tag(list):
tagname = ''
def __init__(self, *arg, **kw):
self.attributes = kw
if self.tagname:
name = self.tagname
self.isSeq = False
else:
name = 'sequence'
self.isSeq = True
self.id = kw.get('id', name)
# self.extend(arg)
self.indentlvl = 0
for a in arg:
self.addObj(a)
def __iadd__(self, obj):
"重载累加操作符"
if isinstance(obj, Tag) and obj.isSeq:
for o in obj:
self.addObj(o)
else:
self.addObj(obj)
return self
def addObj(self, obj):
if not isinstance(obj, Tag):
# obj = str(obj)
# obj=unicode(obj) python3 不支持该函数，所以在此将其注释
pass
id = self.setID(obj)
setattr(self, id, obj)
self.append(obj)
def setID(self, obj):
if isinstance(obj, Tag):
id = obj.id
n = len(
[t for t in self if isinstance(t, Tag) and t.id.startswith(id)])
else:
id = 'content'
n = len([t for t in self if not isinstance(t, Tag)])
if n:
id = '%s_%03i' % (id, n)
if isinstance(obj, Tag):
obj.id = id
return id
def __add__(self, obj):
if self.tagname:
return Tag(self, obj)
self.addObj(obj)
return self
def __lshift__(self, obj):
"操作符<<重载"
self += obj
if isinstance(obj, Tag):
return obj
# 添加一个文件参数，可以输出代码段到指定文件，因此可以不必生成完整的页面
def render(self, file_name=None):
result = ''
if self.tagname:
result = '<%s%s%s>\n' % (
self.tagname, self.renderAtt(), self.selfClose() * ' /')
if not self.selfClose():
for c in self:
if isinstance(c, Tag):
result += c.render()
else:
result += c
if self.tagname:
result += '\n</%s>' % self.tagname
result += '\n'
if file_name:
with open(file_name, "w") as f:
f.write(result)
else:
return result
def renderAtt(self):
result = ''
for n, v in self.attributes.items():
if n != 'txt' and n != 'open':
if n == 'cl':
n = 'class'
result += ' %s="%s"' % (n, v)
return result
def selfClose(self):
return self.tagname in selfClose
def TagFactory(name):
class f(Tag):
tagname = name
f.__name__ = name
return f
thisModule = modules[__name__]
for t in tags:
setattr(thisModule, t, TagFactory(t))
def ValidW3C():
out = a(img(src='http://www.w3.org/Icons/valid-xhtml10',
alt='Valid XHTML 1.0 Strict'), href='http://validator.w3.org/check?uri=referer')
return out
class PyH(Tag):
tagname = 'html'
def __init__(self, name='MyPyHPage'):
self += head()
self += body()
self.attributes = dict(xmlns='http://www.w3.org/1999/xhtml', lang='en')
self.head += title(name)
def __iadd__(self, obj):
if isinstance(obj, head) or isinstance(obj, body):
self.addObj(obj)
elif isinstance(obj, meta) or isinstance(obj, link):
self.head += obj
else:
self.body += obj
id = self.setID(obj)
setattr(self, id, obj)
return self
def addJS(self, *arg):
for f in arg:
self.head += script(type='text/javascript', src=f)
def addCSS(self, *arg):
for f in arg:
self.head += link(rel='stylesheet', type='text/css', href=f)
def addStyleSnippet(self, filename):
"从某个文件中导入css代码段"
with open(filename, "r") as f:
txt = f.read()
self.head += style(txt, type="text/css")
def addScriptSnippet(self, filename):
"从某个文件中导入js代码段"
with open(filename, "r") as f:
txt = f.read()
self.head += script(txt, type="text/javascript")
def printOut(self, file='',encodetype="utf-8", file_operation="w"):#添加一个默认参数用来设置编码方式
if file:
f = open(file, file_operation) #添加一个文件操作参数，实际使用过程中迭代page时不一定每次都要'w'覆盖，有时需要'a'
else:
f = stdout
f.write(doctype)
# f.write(unicode(self.render()).encode(encodetype))
f.write(self.render().encode(encodetype))
f.flush()
if file:
f.close()