#!/usr/bin/python
# -*- coding: utf-8 -*-
import pwb
import pywikibot
import sys
import codecs
import json
import re

try:
	f2 = codecs.open('CSV_OSM.txt', 'ab', 'utf8')
except (OSError, IOError) as e:
	print u'Problemes per obrir l\'arxiu %s' % arxiu
	exit(0)
#"ARTICLEVIQUIPEDIA" P402 "NUMIDENTIFCADOROSM" S143 Q936

liniesCA = []
liniesES = []
liniesFR = []
liniesOC = []
liniesAltres = []
elementSenseWikipedia = []

def main():
	try:
		f = codecs.open("JSONOSM.json", 'r', 'utf8')
	except (OSError, IOError) as e:
		print u'Problemes per obrir l\'arxiu %s' % arxiu
		exit(0)
	txt = f.read()
	f.close()
	jsonObject = json.loads(txt)
	for item in jsonObject['elements']:
		identificador = ""
		article = ""
		element = ""
		auxiliar = ""
		linia = ""
		if item['type'] == "relation":
			identificador = item['id']
			if identificador:
				if item['tags']:
					try:
						article = item['tags']['wikipedia']
					except:
						element = item['tags']['name']
						print u"L\'element %s no té el tag Wikipedia" % (element)
						elementaux = u"%s\t" % (element)
						elementSenseWikipedia.append(elementaux)
					if article:
						llengua = re.findall(u'(.*?):', article)
						auxiliar = re.sub(u'(.*?):', '', article)
						#auxiliar = re.findall(u'ca\:(.*?)', article)
						linia = u"%s\tP402\t\"%s\"\tS143\tQ936\n" % (auxiliar, identificador)
						if llengua[0] == 'ca':
							liniesCA.append(linia)
						elif llengua[0] == 'es':
							liniesES.append(linia)
						elif llengua[0] == 'fr':
							liniesFR.append(linia)
						elif llengua[0] == 'oc':
							liniesOC.append(linia)
						else:
							liniesAltres.append(linia)
						llengua = ""

def escriuArxiuFinal(liniesLlengua, llengua):
	entradetes = u"Aquests son els enllaços %s:\n\n" % (llengua)
	f2.write(entradetes)
	for liniaXX in liniesLlengua:
		f2.write(liniaXX)
	f2.write("\n\n")

if __name__ == '__main__':
	main()
	escriuArxiuFinal(liniesCA, u"català")
	escriuArxiuFinal(liniesES, u"castellà")
	escriuArxiuFinal(liniesFR, u"francès")
	escriuArxiuFinal(liniesOC, u"occità")
	escriuArxiuFinal(liniesAltres, u"Altres")
	escriuArxiuFinal(elementSenseWikipedia, u"sense correspondència")
	f2.close()