# Path: fooltrader/api/technical.py
# def get_security_list(security_type='stock', exchanges=None, start_code=None, end_code=None,
#                       mode='simple', start_list_date=None, codes=None):
#     """
#     get security list.
# 
#     Parameters
#     ----------
#     security_type : str
#         {‘stock’, 'future'},default: stock
#     exchanges : str or list
#         ['sh', 'sz','nasdaq','nyse','amex','shfe','dce','zce'],default: ['sh','sz']
#     start_code : str
#         the start code,work with end,default:None
#         if using codes,it would be ignored
#     end_code : str
#         the end code,works with start,default:None
#         if using codes,it would be ignored
#     mode : str
#         whether parse more security info,{'simple','es'},default:'simple'
#     start_list_date : Timestamp str or Timestamp
#         the filter for start list date,default:None
#     codes : list
#         the exact codes to query,default:None
# 
#     Returns
#     -------
#     DataFrame
#         the security list
# 
#     """
#     df = pd.DataFrame()
#     if type(exchanges) == str:
#         exchanges = [exchanges]
# 
#     if not exchanges:
#         exchanges = SECURITY_TYPE_MAP_EXCHANGES[security_type]
# 
#     if security_type == 'index':
#         df = df.append(pd.DataFrame(CHINA_STOCK_SH_INDEX), ignore_index=True)
#         df = df.append(pd.DataFrame(CHINA_STOCK_SZ_INDEX), ignore_index=True)
#         df = df.append(pd.DataFrame(USA_STOCK_NASDAQ_INDEX), ignore_index=True)
#     else:
#         for exchange in exchanges:
#             the_path = get_security_list_path(security_type, exchange)
#             if os.path.exists(the_path):
#                 converters = None
# 
#                 if mode == 'es' and security_type == 'stock':
#                     converters = {'code': str,
#                                   'sinaIndustry': convert_to_list_if_need,
#                                   'sinaConcept': convert_to_list_if_need,
#                                   'sinaArea': convert_to_list_if_need}
#                 if converters:
#                     df = df.append(pd.read_csv(the_path, converters=converters), ignore_index=True)
#                 else:
#                     df = df.append(pd.read_csv(the_path, dtype=str), ignore_index=True)
# 
#     if not df.empty > 0:
#         df = df_for_date_range(df, start_date=start_list_date)
# 
#         df = df.set_index(df['code'], drop=False)
#         df = df.sort_index()
# 
#         if codes:
#             df = df[df["code"].isin(codes)]
#         elif start_code and end_code:
#             df = df[(df["code"] >= start_code) & (df["code"] <= end_code)]
# 
#         if security_type != 'cryptocurrency':
#             df = df.drop_duplicates(subset='code', keep='last')
# 
#     return df
# 
# Path: fooltrader/contract/files_contract.py
# def get_security_list_path(security_type, exchange):
#     return os.path.join(settings.FOOLTRADER_STORE_PATH, security_type, '{}.csv'.format(exchange))
# 
# Path: fooltrader/utils/utils.py
# def get_exchange(code):
#     if code >= '333333':
#         return 'sh'
#     else:
#         return 'sz'


import json
import threading
import demjson
import scrapy
from scrapy import Request
from scrapy import signals
from fooltrader.api.technical import get_security_list
from fooltrader.contract.files_contract import get_security_list_path
from fooltrader.utils.utils import get_exchange
            self.file_lock.acquire()
            if get_exchange(category['code']) == 'sh':
                df = self.sh_df
            elif get_exchange(category['code']) == 'sz':
                df = self.sz_df

            if category['code'] in df.index:
                current_ind = df.at[category['code'], self.category_type]

                if type(current_ind) == list and (response.meta['ind_name'] not in current_ind):
                    current_ind.append(response.meta['ind_name'])

                elif type(current_ind) == str and response.meta['ind_name'] != current_ind:
                    current_ind = [current_ind, response.meta['ind_name']]
                else:
                    current_ind = response.meta['ind_name']

                df.at[category['code'], self.category_type] = current_ind
            self.file_lock.release()

    @classmethod
    def from_crawler(cls, crawler, *args, **kwargs):
        spider = super(SinaCategorySpider, cls).from_crawler(crawler, *args, **kwargs)
        crawler.signals.connect(spider.spider_closed, signal=signals.spider_closed)
        return spider

    def spider_closed(self, spider, reason):
        if self.sh_df[self.category_type].any():
            self.sh_df.to_csv(get_security_list_path('stock', 'sh'), index=False)
        if self.sz_df[self.category_type].any():

# Path: fooltrader/api/technical.py
# def get_security_list(security_type='stock', exchanges=None, start_code=None, end_code=None,
#                       mode='simple', start_list_date=None, codes=None):
#     """
#     get security list.
# 
#     Parameters
#     ----------
#     security_type : str
#         {‘stock’, 'future'},default: stock
#     exchanges : str or list
#         ['sh', 'sz','nasdaq','nyse','amex','shfe','dce','zce'],default: ['sh','sz']
#     start_code : str
#         the start code,work with end,default:None
#         if using codes,it would be ignored
#     end_code : str
#         the end code,works with start,default:None
#         if using codes,it would be ignored
#     mode : str
#         whether parse more security info,{'simple','es'},default:'simple'
#     start_list_date : Timestamp str or Timestamp
#         the filter for start list date,default:None
#     codes : list
#         the exact codes to query,default:None
# 
#     Returns
#     -------
#     DataFrame
#         the security list
# 
#     """
#     df = pd.DataFrame()
#     if type(exchanges) == str:
#         exchanges = [exchanges]
# 
#     if not exchanges:
#         exchanges = SECURITY_TYPE_MAP_EXCHANGES[security_type]
# 
#     if security_type == 'index':
#         df = df.append(pd.DataFrame(CHINA_STOCK_SH_INDEX), ignore_index=True)
#         df = df.append(pd.DataFrame(CHINA_STOCK_SZ_INDEX), ignore_index=True)
#         df = df.append(pd.DataFrame(USA_STOCK_NASDAQ_INDEX), ignore_index=True)
#     else:
#         for exchange in exchanges:
#             the_path = get_security_list_path(security_type, exchange)
#             if os.path.exists(the_path):
#                 converters = None
# 
#                 if mode == 'es' and security_type == 'stock':
#                     converters = {'code': str,
#                                   'sinaIndustry': convert_to_list_if_need,
#                                   'sinaConcept': convert_to_list_if_need,
#                                   'sinaArea': convert_to_list_if_need}
#                 if converters:
#                     df = df.append(pd.read_csv(the_path, converters=converters), ignore_index=True)
#                 else:
#                     df = df.append(pd.read_csv(the_path, dtype=str), ignore_index=True)
# 
#     if not df.empty > 0:
#         df = df_for_date_range(df, start_date=start_list_date)
# 
#         df = df.set_index(df['code'], drop=False)
#         df = df.sort_index()
# 
#         if codes:
#             df = df[df["code"].isin(codes)]
#         elif start_code and end_code:
#             df = df[(df["code"] >= start_code) & (df["code"] <= end_code)]
# 
#         if security_type != 'cryptocurrency':
#             df = df.drop_duplicates(subset='code', keep='last')
# 
#     return df
# 
# Path: fooltrader/contract/files_contract.py
# def get_security_list_path(security_type, exchange):
#     return os.path.join(settings.FOOLTRADER_STORE_PATH, security_type, '{}.csv'.format(exchange))
# 
# Path: fooltrader/utils/utils.py
# def get_exchange(code):
#     if code >= '333333':
#         return 'sh'
#     else:
#         return 'sz'


# Path: fooltrader/spiders/chinastock/sina_category_spider.py
import json
import threading
import demjson
import scrapy
from scrapy import Request
from scrapy import signals
from fooltrader.api.technical import get_security_list
from fooltrader.contract.files_contract import get_security_list_path
from fooltrader.utils.utils import get_exchange

            self.file_lock.acquire()
            if get_exchange(category['code']) == 'sh':
                df = self.sh_df
            elif get_exchange(category['code']) == 'sz':
                df = self.sz_df

            if category['code'] in df.index:
                current_ind = df.at[category['code'], self.category_type]

                if type(current_ind) == list and (response.meta['ind_name'] not in current_ind):
                    current_ind.append(response.meta['ind_name'])

                elif type(current_ind) == str and response.meta['ind_name'] != current_ind:
                    current_ind = [current_ind, response.meta['ind_name']]
                else:
                    current_ind = response.meta['ind_name']

                df.at[category['code'], self.category_type] = current_ind
            self.file_lock.release()

    @classmethod
    def from_crawler(cls, crawler, *args, **kwargs):
        spider = super(SinaCategorySpider, cls).from_crawler(crawler, *args, **kwargs)
        crawler.signals.connect(spider.spider_closed, signal=signals.spider_closed)
        return spider

    def spider_closed(self, spider, reason):
        if self.sh_df[self.category_type].any():
            self.sh_df.to_csv(get_security_list_path('stock', 'sh'), index=False)
        if self.sz_df[self.category_type].any():

            self.sz_df.to_csv(get_security_list_path('stock', 'sz'), index=False)