rainbowCoder/gen.py at master · ManojRayudu/rainbowCoder · 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
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
import time
import md
import os
import diff
import mgr
import json
import re
from datetime import datetime


class GeneratedRainbowCoder(object):
  post_dir = './post'
  disqus = '''
    <div id="disqus_thread"></div>
    <script type="text/javascript">
      /* * * CONFIGURATION VARIABLES: EDIT BEFORE PASTING INTO YOUR WEBPAGE * * */
      var disqus_shortname = 'rainbowcoder'; // required: replace example with your forum shortname

      /* * * DON'T EDIT BELOW THIS LINE * * */
      (function() {
          var dsq = document.createElement('script'); dsq.type = 'text/javascript'; dsq.async = true;
          dsq.src = '//' + disqus_shortname + '.disqus.com/embed.js';
          (document.getElementsByTagName('head')[0] || document.getElementsByTagName('body')[0]).appendChild(dsq);
      })();
    </script>
    <noscript>Please enable JavaScript to view the <a href="https://disqus.com/?ref_noscript">comments powered by Disqus.</a></noscript>
  '''

  def __init__(self):
    self.cfg = mgr.cfg
    self.pattern = re.compile('"?(post/.*\.md)"?')
    self.idiff = diff.PostDiff()

  def _get_post(self, file_name):
    m = self.pattern.match(file_name)
    if m:
      name = m.groups()[0]
      return name.decode('string_escape')
    else:
      return False


  def _get_head_md(self):
    handle = open(self.cfg['head_md'], 'r')
    data = handle.read()
    handle.close()
    return data.decode('UTF-8')

  def _get_css(self):
    css = self.cfg['css']
    return '<link rel="stylesheet" type="text/css" href="%s">' % (css)

  def _get_filename(self, path):
    base = os.path.basename(path)
    return os.path.splitext(base)[0].decode('UTF-8')

  def _get_file(self, file):
    handle = open(file, 'r')
    data = handle.read()
    handle.close()
    return data.decode('UTF-8')

  def _save_file(self, file, str):
    handle = open(file, 'w')
    handle.write(str.encode('UTF-8'))
    handle.close()

  def _get_title(self, title):
    if title:
      return "<title>"+title+"</title>"
    return ""

  def _get_html_filename(self, md_file):
    title = self._get_filename(md_file)
    return "html/" + title + ".html"


  def get_html(self, md_file, title=None, suffix="", has_disqus=False):
    title = title or self._get_filename(md_file)
    conv = md.MarkDownConvert(self._get_file(md_file) + suffix)

    html =  u'<!DOCTYPE html>'
    html += '<html><head><meta charset="utf-8">'
    html += self._get_css()
    html += self._get_title(title)
    html += "</head><body>"
    html += conv.md2html().decode('UTF-8')
    if has_disqus:
      html += self.disqus
    html += "</body>"
    html += "</html>"
    return html

  def _gen_post(self, md_file, title=None, suffix="", has_disqus=False):
    print "building: " + md_file
    html = self.get_html(md_file, title, suffix, has_disqus)
    html_file = self._get_html_filename(md_file)
    self._save_file(html_file, html)
    return html_file

  def get_post_list(self):
    ret = []
    list_post = os.listdir(self.post_dir)
    for line in list_post:
      md_file = os.path.join(self.post_dir, line)
      if os.path.isfile(md_file):
        title = self._get_filename(md_file)
        if title != 'index':
          last_time = self.idiff.diff_last_time(md_file)
          re_time = re.match("\s*(.*?)\s*\+", last_time).groups()[0]
          struct_time = time.strptime(re_time)
          ret.append({
              'title': title,
              'md_file': md_file,
              'html_file': title + ".html",
              'create_date':  struct_time,
            })

    ret.sort(key=lambda item: item['create_date'], reverse = True)
    return ret

  def _format(self, title, html, date):
    # date = datetime.fromtimestamp(date).strftime('%Y-%m-%d %H:%M:%S')
    return '[%s](%s)\n' % (title, html)

  def building_index_md(self):
    post_list = self.get_post_list()
    head = self._get_head_md() + u'\n\n\n'
    for v in post_list:
      head += self._format(v['title'], v['html_file'], v['create_date'])

    index_md = self.cfg['index_md']
    self._save_file(index_md, head)
    self._gen_post(index_md, "rainbowCoder")


  def _wrapper_post(self, md_file):
    date = self.idiff.diff_last_time(md_file)
    date = "\n\n\n-----\n" + date
    self._gen_post(md_file, suffix=date, has_disqus=True)

  def building_all(self):
    post_list = self.get_post_list()
    self.building_index_md()
    for v in post_list:
      self._wrapper_post(v['md_file'])

  def building_post(self, md_file):
    self.building_index_md()
    self._wrapper_post(md_file)

  def building_update(self):
    diff_list = self.idiff.diff_list()

    for v in diff_list:
      file_name = self._get_post(v['file_name'])
      if file_name != False:
        self._wrapper_post(file_name)

    self.building_index_md()