import sys
import xlrd   #read xls files
import numpy as np
import pandas as pd
import os
from random import choice
import csv
import unicodedata
import networkx as nx
from io_functions import nxgraph_to_gdf

fileDir = os.path.join('GEMAS procedencia tabla.xls')

workbook = xlrd.open_workbook(fileDir)
sheet = workbook.sheet_by_index(0)	#Beta


#data frame for sellos

Index=sheet.col_values(0)

del Index[0]
del Index[0]
del Index[0]
del Index[0]
del Index[0]

for i in range(15):
    del Index[-1]

Columns = sheet.row_values(4)

del Columns[0]
del Columns[-1]



Gems = pd.DataFrame(index=Index,columns=Columns)

for i in range(len(Index)):
    for j in range(len(Columns)):
        row=Index[i]
        col=Columns[j]
        if sheet.row_values(5+i)[1+j]==1:
            Gems.loc[row][col]=sheet.row_values(5+i)[1+j]
        if sheet.row_values(5+i)[1+j]==u'':
            Gems.loc[row][col]=0

#NOD=Columns+Index

G=nx.Graph()

G.add_nodes_from(Columns,tipo='gems')
G.add_nodes_from(Index,tipo='origin')

for i in range(len(Index)):
    for j in range(len(Columns)):
        row=Index[i]
        col=Columns[j]
        if Gems.loc[row][col]==1:
            G.add_edge(row,col)

nxgraph_to_gdf(G,'gemmes.gdf',node_attr=['tipo'])
#, node_attr=None, edge_attr=None, giant=False):