OSDN Git Service

Do not display 0
[fukui-no-namari/fukui-no-namari.git] / src / Hage1 / subjecttxtfile.py
1 # Copyright (C) 2006 by Aiwota Programmer
2 # aiwotaprog@tetteke.tk
3 #
4 # This program is free software; you can redistribute it and/or modify
5 # it under the terms of the GNU General Public License as published by
6 # the Free Software Foundation; either version 2 of the License, or
7 # (at your option) any later version.
8 #
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 # GNU General Public License for more details.
13 #
14 # You should have received a copy of the GNU General Public License
15 # along with this program; if not, write to the Free Software
16 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17
18 import re
19 import os.path
20 import codecs
21 import urllib
22 from misc import get_board_subjecttxt_path
23 from misc import get_board_subjecttxt_url
24 from misc import get_board_idx_path
25
26 REG_EXPR = re.compile("(?P<id>.*).dat<>(?P<title>.*)\((?P<res>\d*)\)")
27
28 def expr(line, func):
29     m = REG_EXPR.match(line)
30     if m:
31         id = m.group("id")
32         title = m.group("title")
33         try:
34             res = int(m.group("res"))
35         except:
36             res = 0
37         func(id, title, res)
38
39 def analyze_subjecttxt(subjecttxt, func):
40     lines = subjecttxt.splitlines()
41     for line in lines:
42         expr(line, func)
43     
44 def load_subjecttxt(bbs, board, func):
45     subjecttxt_path = get_board_subjecttxt_path(bbs, board)
46     if not os.path.exists(subjecttxt_path):
47         return
48
49     f = open(subjecttxt_path)
50     subjecttxt_encoded = f.read()
51     f.close()
52
53     analyze_subjecttxt(subjecttxt_encoded.decode("cp932", "replace"), func)
54
55 def get_subjecttxt(bbs, board):
56     u = urllib.urlopen(get_board_subjecttxt_url(bbs, board))
57     subjecttxt_encoded = u.read()
58     info = u.info()
59
60     subjecttxt_path = get_board_subjecttxt_path(bbs, board)
61     
62     basedir = os.path.dirname(subjecttxt_path)
63     if not os.path.isdir(basedir):
64         os.makedirs(basedir)
65
66     f = open(subjecttxt_path, "w")
67     f.write(subjecttxt_encoded)
68     f.close()
69
70     lastmod = None
71     if "Last-Modified" in info:
72         lastmod = info["Last-Modified"]
73         save_board_idx(bbs, board, lastmod)
74
75     return subjecttxt_encoded.decode("cp932", "replace"), lastmod
76
77 def load_board_idx(bbs, board):
78     lastmod = ""
79     boardidxfile = get_board_idx_path(bbs, board)
80     if os.path.exists(boardidxfile):
81         for line in file(boardidxfile):
82             if line.startswith("lastModified="):
83                 lastmod = line[len("lastModified="):].rstrip("\n")
84                 break
85     return lastmod
86
87 def save_board_idx(bbs, board, lastmod):
88     if not lastmod:
89         return
90
91     boardidx_path = get_board_idx_path(bbs, board)
92     basedir = os.path.dirname(boardidx_path)
93     if not os.path.isdir(basedir):
94         os.makedirs(basedir)
95
96     f = file(boardidx_path, "w")
97     f.write("lastModified=" + lastmod + "\n")
98     f.close()