Mercurial > mercurial > hgweb_golang.cgi

comparison src/kaigo/fwgo/fw_tmpl.go @ 57:05f3d51ad966

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
add fwgo.
author pyon@macmini
date Wed, 15 Jul 2020 18:18:24 +0900
parents
children
comparison
equal deleted inserted replaced
56:7396e7407abd 57:05f3d51ad966
1 /*
2 fwgo.go: Futan-Wariai de Go
3
4 Last Change: 2020-07-14 火 15:04:09.
5 */
6 package main
7
8 /*
9 #cgo LDFLAGS: -L. -lxdwapi -static
10 ##### C_SOURCE #####
11 */
12 import "C"
13
14 import (
15 "encoding/csv"
16 "flag"
17 "fmt"
18 "io"
19 "io/ioutil"
20 "log"
21 "os"
22 "regexp"
23 "strings"
24 "time"
25
26 "golang.org/x/text/encoding/japanese"
27 "golang.org/x/text/transform"
28 )
29
30 type Data struct {
31 Hno string
32 Page string
33 Zip string
34 SendTo string
35 }
36
37 func (d *Data) ToCsv() string {
38 s := []string{d.Hno, d.Page, d.Zip, d.SendTo}
39 return strings.Join(s, ",")
40 }
41
42 var (
43 ver = "0.1"
44
45 clean bool
46 skip bool
47 debug bool
48 xdwfile string
49
50 txtfile = "fwtmp_txt.txt"
51 infofile = "fwtmp_info.csv"
52 orderfile = "order.csv" // input
53 pagefile1 = "fwtmp_page1.txt" // send to sisetsu
54 pagefile2 = "fwtmp_page2.txt" // send to home
55 output1 = "output1.xdw"
56 output2 = "output2.xdw"
57 )
58
59 func init() {
60 /* INITIALIZE FLAGS */
61 flag.BoolVar(&clean, "c", false, "clean temporary files & exit")
62 flag.BoolVar(&skip, "e", false, "use existed files")
63 flag.BoolVar(&debug, "d", false, "debug mode")
64 flag.StringVar(&xdwfile, "i", "KBPV016G.xdw", "target xdw file")
65 }
66
67 func main() {
68 flag.Parse()
69
70 /* PRINT HEADER */
71 fmt.Println("=================================================")
72 fmt.Println(" 負担割合証を... ")
73 fmt.Printf(" - fwgo [ver %s] -\n", ver)
74 fmt.Println("=================================================\n")
75 print_time("now")
76 fmt.Println("[0] start ...")
77
78 /* CLEAN TEMPORARY DIRECTORY */
79 os.Remove(output1)
80 os.Remove(output2)
81 if !skip {
82 clean_full()
83 }
84 if clean {
85 os.Exit(0)
86 }
87 print_time("check done")
88
89 fmt.Println("[1] extract ...")
90 if !skip {
91 C.xdw2txt(C.CString(xdwfile), C.CString(txtfile))
92 }
93 print_time("extract done.")
94
95 c := make(chan int)
96 fmt.Println("[2] split ... (run background)")
97 go func() {
98 if !skip {
99 C.xdwsplit1(C.CString(xdwfile))
100 }
101 print_time("split done.")
102 c <- 1
103 }()
104
105 fmt.Println("[3] analize ...")
106 data_hash, err := analize(txtfile, infofile)
107 if err != nil {
108 log.Fatal(err)
109 }
110 print_time("analize done.")
111 debug_print(debug, fmt.Sprintf("len = %d", len(data_hash)))
112
113 fmt.Println("[4] read order ...")
114 h_order, n, err := read_order(orderfile)
115 if err != nil {
116 log.Fatal(err)
117 }
118 print_time("read order done.")
119 debug_print(debug, fmt.Sprintf("len = %d / %d", len(h_order), n))
120
121 fmt.Println("[5] make_list ...")
122 n, m, l, err := make_list(h_order, data_hash, pagefile1, pagefile2)
123 if err != nil {
124 log.Fatal(err)
125 }
126 print_time("make list done.")
127 debug_print(debug, fmt.Sprintf("order, kunai, kugai = %d, %d, %d", n, m, l))
128
129 <-c
130
131 fmt.Println("[6] merge ...")
132 clean_mini()
133 C.xdwmerge(C.CString(pagefile1), C.CString(output1))
134 print_time("merge 1/2 done.")
135 clean_mini()
136 C.xdwmerge(C.CString(pagefile2), C.CString(output2))
137 print_time("merge 2/2 done.")
138
139 fmt.Println("[7] page ...")
140 C.xdwaddpage(C.CString(output1), C.int(1), C.int(0))
141 print_time("page 1/2 done.")
142 C.xdwaddpage(C.CString(output2), C.int(n + 1), C.int(1))
143 print_time("page 2/2 done.")
144
145 if !debug {
146 fmt.Println("[9] clean ...")
147 clean_full()
148 print_time("clean done.")
149 }
150 }
151
152 func analize(txtfile, infofile string) (map[string]Data, error) {
153 hash := make(map[string]Data)
154
155 c, err := ioutil.ReadFile(txtfile)
156 if err != nil {
157 return hash, err
158 }
159 r := strings.NewReader(string(c))
160 tr := transform.NewReader(r, japanese.ShiftJIS.NewDecoder())
161 b, err := ioutil.ReadAll(tr)
162 if err != nil {
163 return hash, err
164 }
165
166 regHid := regexp.MustCompile(`05[0-9]{4}0[1238][0-9]{8}`)
167 regZip := regexp.MustCompile(`[０１２３４５６７８９]{3}－[０１２３４５６７８９]{4}`)
168
169 var csv string
170
171 buf := strings.Split(string(b), "生年月日年月日")
172 for p, v := range buf[:len(buf)-1] {
173 hid := regHid.FindString(v)
174 hno := hid[6:16]
175
176 zips := regZip.FindAllString(v, -1)
177 if len(zips) != 2 {
178 fmt.Println("warnig:", hno, "have not 2 zip-pattern")
179 }
180
181 s := strings.Split(v, zips[1])
182 s = strings.Split(s[len(s)-1], " 様 ")
183 send := strings.ReplaceAll(s[0], " ", "")
184
185 page := fmt.Sprintf("%05d", p + 1)
186
187 data := Data {
188 Hno: hno,
189 Page: page,
190 Zip: zips[1],
191 SendTo: send,
192 }
193 hash[hno] = data
194
195 csv += data.ToCsv() + "\n"
196 }
197
198 if err := ioutil.WriteFile(infofile, []byte(csv), 0644); err != nil {
199 return hash, err
200 }
201 return hash, nil;
202 }
203
204 func read_order(csvfile string) ([]string, int, error) {
205 var order []string
206
207 c, err := ioutil.ReadFile(csvfile)
208 if err != nil {
209 return order, -1, err
210 }
211 r := strings.NewReader(string(c))
212 tr := transform.NewReader(r, japanese.ShiftJIS.NewDecoder())
213 b, err := ioutil.ReadAll(tr)
214 if err != nil {
215 return order, -1, err
216 }
217
218 n := 0
219 cr := csv.NewReader(strings.NewReader(string(b)))
220 for {
221 record, err := cr.Read()
222 if err == io.EOF {
223 break
224 }
225 if err != nil {
226 return order, n, err
227 }
228
229 if strings.HasPrefix(record[0], "0") {
230 order = append(order, record[0])
231 }
232 n++
233 }
234 return order, n, nil
235 }
236
237 func make_list(h_order []string, hash map[string]Data, pagefile1, pagefile2 string) (int, int, int, error) {
238 var n, m, l int
239 var list1, list2 []string
240 done := make(map[string]bool)
241
242 for _, h := range h_order {
243 if _, ok := done[h]; !ok {
244 if data, ok := hash[h]; ok {
245 list1 = append(list1, data.Page)
246 done[h] = true
247 n++
248 }
249 }
250 }
251
252 for _, z := range []string{"０１４－００", "０１４－０１", "０１４－０８", "０１４－１４１３"} {
253 for _, data := range hash {
254 h := data.Hno
255 if _, ok := done[h]; !ok {
256 if strings.HasPrefix(data.Zip, z) {
257 list2 = append(list2, data.Page)
258 done[h] = true
259 m++
260 }
261 }
262 }
263 }
264
265 for _, data := range hash {
266 h := data.Hno
267 if _, ok := done[h]; !ok {
268 list2 = append(list2, data.Page)
269 done[h] = true
270 l++
271 }
272 }
273
274 if err := write_pagefile(pagefile1, list1); err != nil {
275 return n, m, l, err
276 }
277 if err := write_pagefile(pagefile2, list2); err != nil {
278 return n, m, l, err
279 }
280 return n, m, l, nil
281 }
282
283 func write_pagefile(file string, list []string) error {
284 f, err := os.OpenFile(file, os.O_RDWR|os.O_CREATE, 0755)
285 if err != nil {
286 return err
287 }
288 for _, p := range list {
289 fmt.Fprintf(f, "fwtmp_%s.xdw\n", p)
290 }
291 if err := f.Close(); err != nil {
292 return err
293 }
294 return nil
295 }
296
297 func print_time(msg string) {
298 now := time.Now()
299 fmt.Printf("\t%v # %s\n", now, msg)
300 }
301
302 func debug_print(debug bool, msg string) {
303 if debug {
304 fmt.Printf("\t%s\n", msg)
305 }
306 }
307
308 func clean_full() error {
309 return clean_file("fwtmp_")
310 }
311
312 func clean_mini() error {
313 return clean_file("fwtmp_b")
314 }
315
316 func clean_file(prefix string) error {
317 files, err := ioutil.ReadDir(".")
318 if err != nil {
319 return err
320 }
321
322 for _, file := range files {
323 if strings.HasPrefix(file.Name(), prefix) {
324 os.Remove(file.Name())
325 continue
326 }
327 }
328 return err
329 }
330