108 lines
3.4 KiB
Python
108 lines
3.4 KiB
Python
from django.http import HttpResponse
|
|
import json
|
|
import time
|
|
from datetime import datetime
|
|
|
|
from model import models
|
|
from api.v11 import common
|
|
|
|
|
|
def processRequest(request):
|
|
parsedRequest = common.parseRequest(request)
|
|
|
|
if parsedRequest['error'] is not None:
|
|
return parsedRequest['error']
|
|
|
|
if parsedRequest['response'] is not None:
|
|
return parsedRequest['response']
|
|
|
|
|
|
clientId = parsedRequest["clientId"]
|
|
userId = parsedRequest['userId']
|
|
chartId = request.GET.get('chart', '')
|
|
|
|
|
|
if request.method == 'GET':
|
|
if chartId == '':
|
|
return getAllUserCharts(clientId, userId)
|
|
else:
|
|
return getChartContent(clientId, userId, chartId)
|
|
|
|
elif request.method == 'DELETE':
|
|
if chartId == '':
|
|
return common.error('Wrong chart id')
|
|
else:
|
|
return removeChart(clientId, userId, chartId)
|
|
|
|
elif request.method == 'POST':
|
|
chartName = request.POST.get('name')
|
|
symbol = request.POST.get('symbol')
|
|
resolution = request.POST.get('resolution')
|
|
content = request.POST.get('content')
|
|
if chartId == '':
|
|
return saveChart(clientId, userId, chartName, symbol, resolution, content)
|
|
else:
|
|
return rewriteChart(clientId, userId, chartId, chartName, symbol, resolution, content)
|
|
|
|
else:
|
|
return common.error('Wrong request')
|
|
|
|
|
|
#-----------------------------------------------------------------------------------------------------
|
|
#-----------------------------------------------------------------------------------------------------
|
|
#-----------------------------------------------------------------------------------------------------
|
|
|
|
|
|
def getAllUserCharts(clientId, userId):
|
|
chartsList = models.Chart.objects.filter(ownerSource = clientId, ownerId = userId)
|
|
result = map(lambda x : {'id': x.id, 'name': x.name, 'timestamp': time.mktime(x.lastModified.timetuple()), 'symbol': x.symbol, 'resolution': x.resolution} , chartsList)
|
|
return common.response(json.dumps({'status': "ok", 'data': list(result)}))
|
|
|
|
|
|
def getChartContent(clientId, userId, chartId):
|
|
try:
|
|
chart = models.Chart.objects.get(ownerSource = clientId, ownerId = userId, id = chartId)
|
|
result = json.dumps({'status': 'ok', 'data': { 'id': chart.id, 'name': chart.name, 'timestamp': time.mktime(chart.lastModified.timetuple()), 'content': chart.content}})
|
|
return common.response(result)
|
|
except:
|
|
return common.error('Chart not found')
|
|
|
|
|
|
def removeChart(clientId, userId, chartId):
|
|
try:
|
|
chart = models.Chart.objects.get(ownerSource = clientId, ownerId = userId, id = chartId)
|
|
chart.delete()
|
|
return common.response(json.dumps({'status': 'ok'}))
|
|
except:
|
|
return common.error('Chart not found')
|
|
|
|
|
|
def saveChart(clientId, userId, chartName, symbol, resolution, content):
|
|
newChart = models.Chart(
|
|
ownerSource = clientId,
|
|
ownerId = userId,
|
|
name = chartName,
|
|
content = content,
|
|
lastModified = datetime.now(),
|
|
symbol = symbol,
|
|
resolution = resolution
|
|
)
|
|
|
|
newChart.save()
|
|
return common.response(json.dumps({'status': 'ok', 'id': newChart.id}))
|
|
|
|
|
|
def rewriteChart(clientId, userId, chartId, chartName, symbol, resolution, content):
|
|
try:
|
|
chart = models.Chart.objects.get(ownerSource = clientId, ownerId = userId, id = chartId)
|
|
chart.lastModified = datetime.now()
|
|
chart.content = content
|
|
chart.name = chartName
|
|
chart.symbol = symbol
|
|
chart.resolution = resolution
|
|
|
|
chart.save()
|
|
return common.response(json.dumps({'status': 'ok'}))
|
|
except:
|
|
return common.error('Chart not found')
|