# encoding: utf-8

"""
@author: ouwj
@position: Lab
@file: nvaWordFlag.py
@time: 16-10-12 下午4:50
"""

import os
import codecs
import json


FATHER_DIR_NAME = 'youji/'
FATHER_NVA_DIR_NAME = 'youjiNVA/'

class NVA_WordFlag:
    def __init__(self):
        self.nvaDocs = []
        self.nvaFlags = []
        self.totalFileNum = 0

    def loadJson(self):
        filenames = os.listdir(FATHER_NVA_DIR_NAME)
        filenames = sorted(filenames)
        for filename in filenames:
            print filename
            self.totalFileNum += 1
            fin = codecs.open(FATHER_NVA_DIR_NAME + filename, 'r', 'utf8')
            docs_and_flags = fin.readlines()
            self.nvaDocs.append( json.loads(docs_and_flags[0]) )
            self.nvaFlags.append( json.loads(docs_and_flags[1]) )
            fin.close()

    def getTotalFileNum(self):
        return self.totalFileNum

    def getNVADocs(self):
        return self.nvaDocs

    def getNVAFlags(self):
        return self.nvaFlags