#!/usr/bin/env python
# coding: utf-8

# 「標準地域コード」の助けを借りつつ、「全国地方公共団体コード」の RDF を生成する。

# In[1]:


import pandas as pd
import numpy as np
import rdflib
import datetime
import re
import lxml.html
import rdflib
import math
import os.path
import urllib.parse
import urllib.request
import jaconv
from distutils.version import LooseVersion


# In[2]:


from rdflib.namespace import RDF, RDFS, DCTERMS, XSD, SKOS
SAC = rdflib.Namespace("http://data.e-stat.go.jp/lod/sac/")
SACS = rdflib.Namespace("http://data.e-stat.go.jp/lod/terms/sacs#")
JITI = rdflib.Namespace("http://hkwi.github.com/denshijiti/#")
JITIS = rdflib.Namespace("http://hkwi.github.com/denshijiti/terms#")
IC = rdflib.Namespace("http://imi.go.jp/ns/core/rdf#")


# e-stat「標準地域コード」を取り込む。http://data.e-stat.go.jp/lodw/rdfschema/downloadfile/

# In[3]:


estat = rdflib.Graph()
estat.load("http://data.e-stat.go.jp/lodw/download/rdfschema/StandardAreaCode.ttl", format="turtle")


# 現在の excel を収集する。

# In[4]:


# xmlns:gcse="uri:google-did-not-provide-a-real-ns"
url = "http://www.soumu.go.jp/denshijiti/code.html"
p = lxml.html.parse(urllib.request.urlopen(url), base_url=url)
r = p.getroot()
r.make_links_absolute()
for h in r.xpath("//h3"):
    if h.text.find("都道府県コード") >= 0:
        for l in h.xpath("following-sibling::node()//li"):
            if not l.text or l.text.find("都道府県") < 0:
                continue
            if l.text.find("改正一覧") >= 0:
                for a in l.xpath("self::node()//a/@href"):
                    if a.lower().endswith(".xls"):
                        clist_hist = a
                        break
            else:
                for a in l.xpath("self::node()//a/@href"):
                    if a.lower().endswith(".xls"):
                        clist = a
                        break

assert clist
assert clist_hist
assert clist_hist != clist


# まず過去履歴を集める

# In[5]:


x = pd.read_excel(clist_hist, skiprows=1, header=[0,1,2])
x = pd.read_excel("http://www.soumu.go.jp/main_content/000562731.xls", skiprows=1, header=[0,1,2])
def offset():
    for i,c in enumerate(x.columns):
        for j,e in enumerate(c):
            if "都道府県" in e:
                return i,j

x.columns=["a","b","c","d","e"][:offset()[0]] + [
    "都道府県名",
    "改正前コード","改正前市区町村名","改正前市区町村名ふりがな",
    "改正区分","改正年月日",
    "改正後コード","改正後市区町村名","改正後市区町村名ふりがな",
    "事由等"
]
x.index = range(len(x.index))


# In[6]:


asub = x["改正前市区町村名"].str.extract(r"^[\(（](.*)[\)）]$", expand=False)
bsub = x["改正後市区町村名"].str.extract(r"^[\(（](.*)[\)）]$", expand=False)
ks = ["都道府県名", "改正前コード", "改正区分", "改正年月日", "改正後コード"]
x2 = x[asub.isnull() & bsub.isnull() & x[ks].notnull().any(axis=1)].assign(asub=None, bsub=None)
for i,d in asub[asub.notnull()].iteritems():
    x2.loc[i-1, "asub"] = d
for i,d in asub[bsub.notnull()].iteritems():
    x2.loc[i-1, "bsub"] = d

x2.index = range(len(x2.index))


# In[7]:


for ri,r in x2.iterrows():
    for ci,c in r.iteritems():
        if c=="〃":
            x2.loc[ri, ci] = x2.loc[ri-1, ci]


# In[8]:


cids = []
cid = 0
t = x2.assign(u=lambda o: o["都道府県名"].notnull())
for ri, r in x2.iterrows():
    if not isinstance(r["都道府県名"], float):
        cid += 1
    cids.append(cid)


# In[9]:


dts = []
t = pd.concat([x2, pd.Series(cids, name="cid"), pd.Series(None, name="date")], axis=1)
for i in range(cid):
    u = t[(t["cid"]==i+1) & t["改正年月日"].notnull()]
    assert len(set(u["改正年月日"]))==1, u.to_csv()
    for d in set(u["改正年月日"]):
        if isinstance(d, datetime.datetime):
            dt = d.date()
        elif isinstance(d, datetime.date):
            dt = d
        else:
            m = re.match(r"H(\d{2})\.(\d+)\.(\d+)", d)
            dt = datetime.date(2000+int(m.group(1))-12, int(m.group(2)), int(m.group(3)))
    for ri,r in t[(t["cid"]==i+1)].iterrows():
        t.loc[ri,"date"] = dt


# In[10]:


g = rdflib.Graph()
g.bind("ic", IC)
g.bind("sac", SAC)
g.bind("skos", SKOS)
g.bind("jiti", JITI)
g.bind("jitis", JITIS)
g.bind("dcterms", DCTERMS)

for cls in ("StandardAreaCode", "CodeSet", "CodeChangeEvent"):
    g.add((JITIS[cls], RDF.type, RDFS["Class"]))

g.add((JITIS["StandardAreaCode"], RDFS.subClassOf, IC["住所"]))

def get_code(code_like):
    code = "%06d" % int(code_like)
    assert code_checksum(code) == code[5:], "code %s checksum error" % code
    return code

def code_checksum(code):
    s = np.array([int(s) for s in code[:5]]) * np.array([6,5,4,3,2])
    return str(11 - s.sum() % 11)[-1]

def get_code_id(url_like):
    m = re.search(r"C(?P<code>\d{5})-(?P<ymd>\d{8})$", url_like)
    assert m, "code_id %s format error" % url_like
    return m.group(0)

def sacq(code5):
    return estat.query('''
    PREFIX dcterms: <http://purl.org/dc/terms/>
    PREFIX sacs: <http://data.e-stat.go.jp/lod/terms/sacs#>
    PREFIX rdfs: <http://www.w3.org/2000/01/rdf-schema#>
    SELECT ?s ?n ?d WHERE {
        ?s a sacs:StandardAreaCode ;
            rdfs:label ?n ;
            dcterms:identifier "%s" ;
            dcterms:issued ?d .
        FILTER ( LANG(?n) = "ja" )
    }
    ORDER BY DESC(?d)
    ''' % code5)

class Code(object):
    codes = {}
    @classmethod
    def singleton(cls, code_id):
        obj = cls.codes.get(code_id)
        if obj is None:
            cls.codes[code_id] = obj = cls(code_id)
        return obj
    
    def __init__(self, code_id):
        m = re.match(r"C(?P<code>\d{5})-(?P<ymd>\d{8})$", code_id)
        assert m, "code %s format error" % code_id
        self.code = m.group("code")
        self.csum = code_checksum(self.code)
        self.ymd = datetime.datetime.strptime(m.group("ymd"), "%Y%m%d").date()
        
        # e-Stat LOD ベースで採番する
        self.sac = JITI[code_id]
        g.add((self.sac, RDF["type"], JITIS["StandardAreaCode"]))
        g.add((self.sac, SKOS["closeMatch"], SAC[code_id]))
        g.add((self.sac, DCTERMS["identifier"], rdflib.Literal(self.code)))
        g.add((self.sac, DCTERMS["issued"], rdflib.Literal(self.ymd.strftime("%Y-%m-%d"), datatype=XSD.date)))
        # 6桁コードでリンクできるようにする
        g.add((self.sac, JITIS["code"], rdflib.Literal(self.code+self.csum)))
        
        # 共通語彙基盤 住所IEP : 現状は blank node での出力を強制する
        pref = rdflib.BNode()
        g.add((pref, RDF.type, IC["コード型"]))
        g.add((pref, IC["識別値"], rdflib.Literal(self.code[:2], datatype=XSD.string)))
        g.add((self.sac, IC["都道府県コード"], pref))
        
        ward = rdflib.BNode()
        g.add((ward, RDF.type, IC["コード型"]))
        g.add((ward, IC["識別値"], rdflib.Literal(self.code[2:]+self.csum, datatype=XSD.string)))
        g.add((self.sac, IC["市区町村コード"], ward))

    def set_name(self, name, kana):
        assert name != "同左"
        assert kana != "同左"
        name = name.strip()
        kana = kana.strip()
        # e-Stat LOD 準拠
        g.add((self.sac, RDFS["label"], rdflib.Literal(name, lang="ja")))
        g.add((self.sac, RDFS["label"], rdflib.Literal(kana, lang="ja-hrkt")))
        
        ku = None
        sub = int(self.code[2:])
        if self.code[2:] == "000":
            g.add((self.sac, JITIS["type"], JITIS["都道府県"]))
        elif self.code[:3] == "131":
            g.add((self.sac, JITIS["type"], JITIS["特別区"]))
        elif self.code[2:] == "100":
            g.add((self.sac, JITIS["type"], JITIS["指定都市"]))
        elif self.code[2] == "1":
            if name.endswith("市"):
                g.add((self.sac, JITIS["type"], JITIS["指定都市"]))
            else:
                g.add((self.sac, JITIS["type"], JITIS["指定都市の区"]))
                for i in range(sub // 10 * 10, 90, -10):
                    if ku:
                        break
                    for s,n,d in sacq("%s%03d" % (self.code[:2], i)):
                        if d.value <= self.ymd:
                            if n.endswith("市"):
                                ku = n
                                break
        elif self.code[2] == "2":
            g.add((self.sac, JITIS["type"], JITIS["市"]))
        elif self.code[:2] == "01":
            if sub % 30:
                gr = "01%03d" % (sub // 30 * 30)
                g.add((self.sac, JITIS["group"], rdflib.Literal(gr + code_checksum(gr))))
        elif self.code[:2] == "47":
            if 340 < sub and sub < 370:
                gr = "47340"
                g.add((self.sac, JITIS["group"], rdflib.Literal(gr + code_checksum(gr))))
            elif 370 < sub and sub < 380:
                gr = "47370"
                g.add((self.sac, JITIS["group"], rdflib.Literal(gr + code_checksum(gr))))
            elif sub % 20:
                gr = "47%03d" % (sub // 20 * 20)
                g.add((self.sac, JITIS["group"], rdflib.Literal(gr + code_checksum(gr))))
        elif sub % 20:
            gr = "%s%03d" % (self.code[:2], sub // 20 * 20)
            g.add((self.sac, JITIS["group"], rdflib.Literal(gr + code_checksum(gr))))

        if self.code[2:] == "000":
            g.add((self.sac, IC["都道府県"], rdflib.Literal(name, lang="ja")))
        else:
            if ku:
                g.add((self.sac, IC["市区町村"], ku))
                g.add((self.sac, IC["区"], rdflib.Literal(name, lang="ja")))
            else:
                g.add((self.sac, IC["市区町村"], rdflib.Literal(name, lang="ja")))
            for s,n,d in sacq(self.code[:2]+"000"):
                if d.value <= self.ymd:
                    g.add((self.sac, IC["都道府県"], n))
                    break

code_ids = []

last_date = None

ev = None
cid = -1
for ri, r in t.sort_values(["date","cid"]).iterrows():
    last_date = r["date"]
    
    if cid != r["cid"]:
        dtstr = r["date"].strftime("%Y-%m-%d")
        ev = rdflib.BNode("_:ev%s-%02d" % (dtstr,cid))
        g.add((ev, RDF["type"], JITIS["CodeChangeEvent"]))
        g.add((ev, DCTERMS["issued"], rdflib.Literal(dtstr, datatype=XSD.date)))
        cid = r["cid"]
    
    if not math.isnan(r["改正前コード"]):
        tc = get_code(r["改正前コード"])
        cids = [cid for c,cid in code_ids if c==tc]
        code_id = None
        if cids:
            code_id = cids[-1]
        else:
            for s,n,d in sacq(tc[:5]):
                if d.value < r["date"]:
                    code_id = get_code_id(s)
                    break
        assert code_id
        g.add((ev, JITIS["old"], JITI[code_id]))
        Code.singleton(code_id).set_name(r["改正前市区町村名"], r["改正前市区町村名ふりがな"])
    
    code = None
    code_id = None
    if r["改正後コード"]=="削除" or r["改正区分"] == "欠番":
        pass
    elif r["改正後コード"] == "同左":
        if math.isnan(r["改正前コード"]):
            raise Exception(r.to_csv())
        code = get_code(r["改正前コード"])
        code_id = "C%s-%s" % (code[:5], r["date"].strftime("%Y%m%d"))
    else:
        try:
            code = get_code(r["改正後コード"])
            code_id = "C%s-%s" % (code[:5], r["date"].strftime("%Y%m%d"))
        except:
            raise Exception(r.to_csv())
    
    if code and code_id:
        for t in g.triples((JITI[code_id], RDFS["label"], None)):
            g.remove(t)
        
        code_ids.append((code, code_id))
        g.add((ev, JITIS["new"], JITI[code_id]))
        
        name = r["改正後市区町村名"]
        if name.strip()=="同左":
            name = r["改正前市区町村名"]
        
        kana = r["改正後市区町村名ふりがな"]
        if isinstance(kana, float) or kana.strip()=="同左":
            kana = r["改正前市区町村名ふりがな"]
            # errata in xls row 1319
            if isinstance(kana, float) and name=="西海市":
                kana = "さいかいし"
        
        Code.singleton(code_id).set_name(name, kana)


# コードセットの登録。

# In[11]:


cs = []

pq = '''
PREFIX dcterms: <http://purl.org/dc/terms/>
SELECT ?s WHERE {
  ?s a <%s> ;
     dcterms:identifier "%s" ;
     dcterms:issued ?d .
}
ORDER BY DESC(?d)
'''

x = pd.read_excel(clist)
for ri,r in x.iterrows():
    code = get_code(r["団体コード"])
    ident = rdflib.Literal(code[:5])
    code_id = None
    for s, in g.query(pq % (JITIS["StandardAreaCode"], ident)):
        code_id = get_code_id(s)
        break
    
    if code_id is None:
        for s, in estat.query(pq % (SACS["StandardAreaCode"], ident)):
            code_id = get_code_id(s)
            break
        
        name = r["市区町村名\n（漢字）"]
        if isinstance(name, float) or not name.strip():
            name = r["都道府県名\n（漢字）"]

        kana = r["市区町村名\n（カナ）"]
        if isinstance(kana, float) or not kana.strip():
            kana = r["都道府県名\n（カナ）"]
        
        assert code_id, code
        Code.singleton(code_id).set_name(name, jaconv.kata2hira(jaconv.h2z(kana)))
    
    assert code_id, code
    cs.append(code_id)

if LooseVersion(pd.__version__) >= LooseVersion("0.21.0"):
    x = pd.read_excel(clist, sheet_name=1, header=None)
else:
    x = pd.read_excel(clist, sheetname=1, header=None)

for ri,r in x.iterrows():
    code = get_code(r[0])
    ident = rdflib.Literal(code[:5])
    code_id = None
    for s, in g.query(pq % (JITIS["StandardAreaCode"], ident)):
        code_id = get_code_id(s)
        break
    
    if code_id is None:
        # 「札幌市中央区」などが入っている。
        # e-Stat では「中央区」が使われていて、揃えるために estat の値を引く
        #name = r[1]
        #kana = r[2]
        name = kana = None
        for s, in estat.query(pq % (SACS["StandardAreaCode"], ident)):
            code_id = get_code_id(s)
            for n in estat.objects(s, RDFS["label"]):
                if n.language=="ja":
                    name = n
                elif n.language=="ja-hrkt":
                    kana = n
        
        Code.singleton(code_id).set_name(name, jaconv.kata2hira(jaconv.h2z(kana)))
    
    assert code_id, code
    cs.append(code_id)


# 過去履歴を逆向きに遡ってコードセットを登録する。

# In[12]:


date = last_date
b = JITI["CS-" + date.strftime("%Y-%m-%d")]
g.add((b, rdflib.RDF["type"], JITIS["CodeSet"]))
g.add((b, DCTERMS["issued"], rdflib.Literal(dtstr, datatype=XSD.date)))
for c in cs:
    g.add((b, DCTERMS["hasPart"], JITI[c]))

    
pq = '''
PREFIX xsd: <http://www.w3.org/2001/XMLSchema#>
PREFIX jitis: <http://hkwi.github.com/denshijiti/terms#>
PREFIX dcterms: <http://purl.org/dc/terms/>
SELECT ?v WHERE {
    ?s a jitis:CodeChangeEvent ;
        dcterms:issued "%s"^^xsd:date ;
        <%s> ?v .
}
'''

dts = g.query('''
PREFIX xsd: <http://www.w3.org/2001/XMLSchema#>
PREFIX jitis: <http://hkwi.github.com/denshijiti/terms#>
PREFIX dcterms: <http://purl.org/dc/terms/>
SELECT DISTINCT ?d WHERE {
    ?s a jitis:CodeChangeEvent ;
        dcterms:issued ?d .
}
ORDER BY DESC(?d)
''')
for dt, in dts:
    if dt.value < date: # 履歴と一覧が同期していないときのための安全
        dtstr = date.strftime("%Y-%m-%d")
        b = JITI["CS-" + dtstr]
        g.add((b, rdflib.RDF["type"], JITIS["CodeSet"]))
        g.add((b, DCTERMS["issued"], rdflib.Literal(dtstr, datatype=XSD.date)))
        for c in cs:
            g.add((b, DCTERMS["hasPart"], JITI[c]))
    date = dt.value
    
    for old, in g.query(pq % (JITIS["old"], dt.value.strftime("%Y-%m-%d"))):
        code_id = get_code_id(old)
        cs.append(code_id)

    for new, in g.query(pq % (JITIS["new"], dt.value.strftime("%Y-%m-%d"))):
        code_id = get_code_id(new)
        try:
            cs.remove(code_id)
        except:
            raise Exception(code_id)


# In[13]:


with open("code.ttl", "wb") as f:
    g.serialize(destination=f, format="turtle")

