views.py

# -*- coding:utf-8 -*-
from django.shortcuts import render
from projectManage.models import *
from accounts.models import *
# from django.core.urlresolvers import reverse
from django.http import HttpResponse, HttpResponseRedirect
from utils.functionTools.generalFunction import noneIfEmptyString,noneIfNoKey,myError
from .search_relation import search_relation, search_genes, search_papers, search_one_sentence, search_three_sentence, search_related_disease
from system.gene import get_gene_info
import json
import string
import random


# Create your views here.

def randomGene(request):
	try:
		body = request.body
		body = body.decode('utf-8')
		data = json.loads(body)
		gene_file = open('data/gene_name.json', 'r')
		gene_list = json.loads(gene_file.read())
		gene_name = random.choice(gene_list)
		search_result = search_relation(gene_name)
		if len(search_result['children']) > 10:
			search_result['children'].sort(key=lambda d:float(d['relations']), reverse=True)
		search_result['children'] = search_result['children'][:10]
		for gene in search_result['children']:
			gene_name = gene['name']
			temp_result = search_relation(gene_name)
			if len(temp_result['children']) > 10:
				temp_result['children'].sort(key=lambda d:float(d['relations']), reverse=True)
				gene['children'] = temp_result['children'][:10]
			else:
				gene['children'] = temp_result['children']
		result = {
			'successful': True,
			'data': search_result,
			'error': {
				'id': '',
				'msg': '',
			},
		}
	except myError as e:
		print(e.value)
		result = {
			'successful': False,
			'error': {
				'id': '3',
				'msg': e.value,
			}
		}
	except Exception as e:
		print(e.args)
		result = {
			'successful': False,
			'error': {
				'id': '1024',
				'msg': e.args
			}
		}
	finally:
		return HttpResponse(json.dumps(result), content_type='application/json')

def searchGenes(request):
	try:
		body = request.body
		body = body.decode('utf-8')
		data = json.loads(body)
		keyword = data['keyword']
		search_result = search_genes(keyword)
		result = {
			'successful': True,
			'data': search_result,
			'error': {
				'id': '',
				'msg': '',
			},
		}
	except myError as e:
		result = {
			'successful': False,
			'error': {
				'id': '3',
				'msg': e.value,
			}
		}
	except Exception as e:
		result = {
			'successful': False,
			'error': {
				'id': '1024',
				'msg': e.args
			}
		}
	finally:
		return HttpResponse(json.dumps(result), content_type='application/json')


def getGeneInfo(request):
	try:
		body = request.body
		body = body.decode('utf-8')
		data = json.loads(body)
		gene_name = data['gene_name']
		gene = Gene.objects.filter(name=gene_name).first()
		gene_id = gene.gene_id
		get_result = get_gene_info(gene_id)
		result = {
			'successful': get_result[0],
			'data': get_result[1],
			'error': {
				'id': '',
				'msg': '',
			},
		}
	except myError as e:
		result = {
			'successful': False,
			'error': {
				'id': '3',
				'msg': e.value,
			}
		}
	except Exception as e:
		result = {
			'successful': False,
			'error': {
				'id': '1024',
				'msg': e.args
			}
		}
	finally:
		return HttpResponse(json.dumps(result), content_type='application/json')

def getRelatedGene(request):
	try:
		body = request.body
		body = body.decode('utf-8')
		data = json.loads(body)
		gene_name = data['gene_name']
		realated_gene_list = []
		realated_genes = search_relation(gene_name)
		if realated_genes:
			if len(realated_genes['children']) > 10:
				print(realated_genes)
				realated_genes['children'].sort(key=lambda d:float(d['relations']), reverse=True)
				realated_genes['children'] = realated_genes['children'][:10]
			print(realated_genes)
		else: 
			realated_genes['main_gene'] = gene_name
			realated_genes['children'] = list()
		result = {
			'successful': True,
			'data': realated_genes,
			'error': {
				'id': '',
				'msg': '',
			},
		}
	except myError as e:
		result = {
			'successful': False,
			'error': {
				'id': '3',
				'msg': e.value,
			}
		}
	except Exception as e:
		result = {
			'successful': False,
			'error': {
				'id': '1024',
				'msg': e.args
			}
		}
	finally:
		return HttpResponse(json.dumps(result), content_type='application/json')

def getRelatedPaper(request):
	try:
		body = request.body
		body = body.decode('utf-8')
		data = json.loads(body)
		gene_name = data['gene_name']
		related_paper_list = search_papers(gene_name)
		result = {
			'successful': True,
			'data': related_paper_list,
			'error': {
				'id': '',
				'msg': '',
			},
		}
	except myError as e:
		result = {
			'successful': False,
			'error': {
				'id': '3',
				'msg': e.value,
			}
		}
	except Exception as e:
		result = {
			'successful': False,
			'error': {
				'id': '1024',
				'msg': e.args
			}
		}
	finally:
		return HttpResponse(json.dumps(result), content_type='application/json')

def getOneSentence(request):
	try:
		body = request.body
		body = body.decode('utf-8')
		data = json.loads(body)
		gene_name_one = data['source_name']
		gene_name_two = data['target_name']
		sentenceList = search_one_sentence(gene_name_one,gene_name_two)
		result = {
			'successful': True,
			'data': sentenceList,
			'error': {
				'id': '',
				'msg': '',
			},
		}
	except myError as e:
		result = {
			'successful': False,
			'error': {
				'id': '3',
				'msg': e.value,
			}
		}
	except Exception as e:
		result = {
			'successful': False,
			'error': {
				'id': '1024',
				'msg': e.args
			}
		}
	finally:
		return HttpResponse(json.dumps(result), content_type='application/json')

def getThreeSentences(request):
	try:
		body = request.body
		body = body.decode('utf-8')
		data = json.loads(body)
		gene_name_one = data['source_name']
		gene_name_two = data['target_name']
		sentenceList = search_three_sentence(gene_name_one,gene_name_two)
		result = {
			'successful': True,
			'data': sentenceList,
			'error': {
				'id': '',
				'msg': '',
			},
		}
	except myError as e:
		result = {
			'successful': False,
			'error': {
				'id': '3',
				'msg': e.value,
			}
		}
	except Exception as e:
		result = {
			'successful': False,
			'error': {
				'id': '1024',
				'msg': e.args
			}
		}
	finally:
		return HttpResponse(json.dumps(result), content_type='application/json')

def getRelatedDisease(request):
	try:
		body = request.body
		body = body.decode('utf-8')
		data = json.loads(body)
		gene_name = data['gene_name']
		realated_disease_list = search_related_disease(gene_name)
		result = {
			'successful': True,
			'data': realated_disease_list,
			'error': {
				'id': '',
				'msg': '',
			},
		}
	except myError as e:
		result = {
			'successful': False,
			'error': {
				'id': '3',
				'msg': e.value,
			}
		}
	except Exception as e:
		result = {
			'successful': False,
			'error': {
				'id': '1024',
				'msg': e.args
			}
		}
	finally:
		return HttpResponse(json.dumps(result), content_type='application/json')