# Without this, we couldn't use the Django environment

import os
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "gargantext_web.settings")
os.environ.setdefault("DJANGO_HSTORE_GLOBAL_REGISTER", "False")

# We're gonna use all the models!

# Django models
from node import models

# SQLA models
from gargantext_web.db import *

# Reset: all data
#
#tables_to_empty = [
#    Node,
#    Node_Hyperdata,
#    Hyperdata,
#    NodeType,
#    ResourceType,
#    Resource,
#]
#for table in tables_to_empty:
#    print('Empty table "%s"...' % (table._meta.db_table, ))
#    table.objects.all().delete()

# Integration: hyperdata types

print('Initialize hyperdata...')
hyperdata = {
    'publication_date': 'datetime',
    'authors': 'string',
    'language_fullname': 'string',
    'abstract': 'text',
    'title': 'string',
    'source': 'string',
    'volume': 'string',
    'text': 'text',
    'page': 'string',
    'doi': 'string',
    'journal': 'string',
}
for name, type_name in hyperdata.items():
    models.Hyperdata(name=name, type=type_name).save()


# Integration: languages

print('Initialize languages...')
import pycountry
Language.objects.all().delete()
for language in pycountry.languages:
    if 'alpha2' in language.__dict__:
        models.Language(
            iso2 = language.alpha2,
            iso3 = language.bibliographic,
            fullname = language.name,
            implemented = 1 if language.alpha2 in ['en', 'fr'] else 0,
        ).save()


# Integration: users
<<<<<<< Updated upstream

print('Initialize users...')
me = models.User.objects.get_or_create(username='alexandre')
gargantua, created = models.User.objects.get_or_create(username='gargantua')
node_root = Node(user_id=gargantua.id, type_id=cache.NodeType['Root'].id, name='Root')
node_stem = Node(user_id=gargantua.id, type_id=cache.NodeType['Stem'].id, name='Stem', parent_id=node_root.id)
node_lem = Node(user_id=gargantua.id, type_id=cache.NodeType['Lem'].id, name='Lem', parent_id=node_root.id)

session.add(node_root)
session.add(node_stem)
session.add(node_lem)
session.commit()

# Integration: node types

print('Initialize node types...')

node_types = [
        'Root', 'Trash',
        'Project', 'Corpus', 'Document',
        'MiamList', 'StopList', 'MainList',
        'Stem', 'Lem', 'Group', 'Tfidf',
        'Cooccurrence', 'WhiteList', 'BlackList'
        ]

for node_type in node_types:
    models.NodeType.objects.get_or_create(name=node_type)

# Integration: resource types

print('Initialize users...')
me = session.query(User).filter(User.username=='alexandre').first()
gargantua = session.query(User).filter(User.username=='gargantua').first()
node_root = Node(user_id=gargantua.id, type_id=cache.NodeType['Root'].id, name='Root')
node_stem = Node(user_id=gargantua.id, type_id=cache.NodeType['Stem'].id, name='Stem', parent_id=node_root.id)
node_lem = Node(user_id=gargantua.id, type_id=cache.NodeType['Lem'].id, name='Lem', parent_id=node_root.id)

session.add(node_root)
session.add(node_stem)
session.add(node_lem)
session.commit()



print('Initialize resource...')

from parsing.parsers_config import parsers

for parser in parsers.keys():
    models.ResourceType.objects.get_or_create(name=parser)




# Instantiante table NgramTag:
f = open("part_of_speech_labels.txt", 'r')

for line in f.readlines():
    name, description = line.strip().split('\t')
    _tag = Tag(name=name, description=description)
    session.add(_tag)
session.commit()

f.close()


exit()