ask.py

#!/usr/bin/env python
from twisted.web.resource import Resource
from twisted.web.util import redirectTo
from twisted.web.template import Element, renderer, renderElement, XMLString
from twisted.python.filepath import FilePath

from data import Ask, Profile, User, Tweet
from data import db
from sessions import SessionManager

import config
import decimal
import definitions
import error
import json
import forms
import pages
import twitter_api

D = decimal.Decimal


def assemble(root):
    root.putChild('ask', Main())
    root.putChild('process_ask', Process())
    root.putChild('create_ask', Create())
    root.putChild('withdraw_ask', Withdraw())
    return root


class Main(Resource):
    def render(self, request):
        print '%srequest.args: %s%s' % (config.color.RED, request.args, config.color.ENDC)

        session_user = SessionManager(request).get_session_user()
        session_response = SessionManager(request).get_session_response()
        
        if session_user['id'] == 0:
            return redirectTo('../', request)
        
        filters = {}
        try:
            filters['status'] = request.args.get('status')[0]
        except:
            filters['status'] = 'pending'

        Page = pages.Ask('%s Ask' % config.company_name, 'ask', filters)
        
        Page.session_user = session_user

        print "%ssession_user: %s%s" % (config.color.BLUE, session_user, config.color.ENDC)
        print "%ssession_response: %s%s" % (config.color.BLUE, session_response, config.color.ENDC)

        SessionManager(request).clearSessionResponse()

        request.write('<!DOCTYPE html>\n')
        return renderElement(request, Page)


class Process(Resource):
    def render(self, request):
        print '%srequest.args: %s%s' % (config.color.RED, request.args, config.color.ENDC)

        session_user = SessionManager(request).get_session_user()
        session_user['action'] = 'process_ask'

        print session_user

        try:
            action = request.args.get('action')[0]
        except:
            return redirectTo('../', request)

        response = {} 
        response['error'] = False
        response['action'] = action
        
        if action == 'create':
            response['ask'] = {
                    'id': 0 
                } 
            response['available_balance'] = session_user['available_balance']
            if session_user['available_balance'] <= 0:
                response['error'] = True

        if action == 'withdraw':
            try:
                ask_id = int(request.args.get('id')[0])
            except:
                return redirectTo('../', request)

            response['ask'] = {
                    'id': ask_id 
                } 

        if action == 'engage':
            try:
                ask_id = int(request.args.get('id')[0])
            except:
                return redirectTo('../', request)

            response['ask'] = {
                    'id': ask_id 
                } 

        return json.dumps(response)


class Create(Resource):
    def render(self, request):
        print '%srequest.args: %s%s' % (config.color.RED, request.args, config.color.ENDC)

        if not request.args:
            return redirectTo('../', request)

        session_user = SessionManager(request).get_session_user()
        session_user['action'] = 'save_order'
        
        twitter_status_id = request.args.get('twitter_status_id')[0]
        niche = request.args.get('niche')[0]
        campaign_type = request.args.get('campaign_type')[0]
        charge = request.args.get('price_per_retweet')[0]
        goal = request.args.get('goal')[0]
        
        response = error.price_per_tweet(request, charge)
        if response['error']:
            return json.dumps(response) 
        
        response = error.goal(request, goal)
        if response['error']:
            return json.dumps(response) 

        timestamp = config.create_timestamp()

        asks = db.query(Ask).filter(Ask.user_id == session_user['id'])
        asks = asks.filter(Ask.status == 'active')

        if asks.count() >= 3:
            response = {}
            response['error'] = True
            response['message'] = 'You can only have 3 active asks on the market.' 
            return json.dumps(response) 

        asks = db.query(Ask).filter(Ask.twitter_status_id == twitter_status_id)
        ask = asks.filter(Ask.status == 'active').first()

        if ask:
            response = {}
            response['error'] = True
            response['message'] = 'You already have this tweet on the market.' 
            return json.dumps(response) 

        client = db.query(Profile).filter(Profile.user_id == session_user['id']).first()
        if client.available_balance < D(charge) * D(goal): 
            response = {}
            response['error'] = True
            response['message'] = "You need %s more in your balance to do this." % (D(charge) * D(goal) - D(client.available_balance)) 
            return json.dumps(response) 

        data = {
            'status': 'active',
            'created_at': timestamp,
            'updated_at': timestamp,
            'twitter_id': session_user['twitter_id'],
            'twitter_status_id': twitter_status_id,
            'user_id': session_user['id'],
            'target': 0,
            'goal': goal,
            'cost': charge,
            'campaign_type': campaign_type, 
            'niche': niche
        }

        new_ask = Ask(data)
        db.add(new_ask)

        client = db.query(Profile).filter(Profile.user_id == session_user['id']).first()
        client.available_balance -= D(charge) * D(goal)
        client.reserved_balance += D(charge) * D(goal)

        db.commit()

        #####################################
        # Need to check if API caused error 
        #####################################
        tweet = twitter_api.get_status(twitter_status_id) 
        data = {
            'twitter_status_id': twitter_status_id,
            'created_at': tweet.created_at.encode('utf-8'),
            'text': tweet.text.encode('utf-8')
        }
        print data
        new_tweet = Tweet(data)
        db.add(new_tweet)

        response = {}
        response['error'] = False
        response['message'] = definitions.MESSAGE_SUCCESS
        response['url'] = '../'

        return json.dumps(response) 


class Withdraw(Resource):
    def render(self, request):
        print '%srequest.args: %s%s' % (config.color.RED, request.args, config.color.ENDC)

        session_user = SessionManager(request).get_session_user()
        session_user['action'] = 'withdraw_ask'

        is_confirmed = request.args.get('is_confirmed')[0]
        if is_confirmed == 'no':
            return json.dumps(dict(response=1, text=definitions.MESSAGE_SUCCESS))
        
        try:
            ask_id = int(request.args.get('ask_id')[0])
        except:
            return redirectTo('../', request)
        
        ask = db.query(Ask).filter(Ask.id == ask_id).first()

        if ask.user_id != session_user['id']:
            return redirectTo('../', request)

        timestamp = config.create_timestamp()
        
        ask.updated_at = timestamp 
        ask.status = 'withdrawn'

        client = db.query(Profile).filter(Profile.user_id == session_user['id']).first()
        client.available_balance += ask.cost * (ask.goal - ask.target)
        client.reserved_balance -= ask.cost * (ask.goal - ask.target)

        db.commit()
        return json.dumps(dict(response=1, text=definitions.MESSAGE_SUCCESS))