bid.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 Bid, Profile, User
from data import db
from sessions import SessionManager

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


def assemble(root):
    root.putChild('bid', Main())
    root.putChild('process_bid', Process())
    root.putChild('create_bid', Create())
    root.putChild('withdraw_bid', 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.Bid('%s Bid' % config.company_name, 'bid', 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):
        response = {'error': True}

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

        response['error'] = False
        response['action'] = action

        if action == 'create':
            try:
                twitter_status_id = int(request.args.get('status_id')[0])
                print status_id
            except:
                twitter_status_id = 0

            response['bid'] = {
                    'twitter_status_id': str(twitter_status_id)
                } 

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

            response['bid'] = {
                    'id': bid_id 
                } 

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

            #bid = db.query(Bid).filter(Bid.id == bid_id).first()
            response['bid'] = {
                    'id': bid_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'] = 'create_bid'
        
        price_per_tweet = request.args.get('price_per_tweet')[0]
        campaign_type = request.args.get('campaign_type')[0]
        niche = request.args.get('niche')[0]
        
        response = error.price_per_tweet(request, price_per_tweet)
        if response['error']:
            return json.dumps(response) 

        bids = db.query(Bid).filter(Bid.user_id == session_user['id'])
        bid = bids.filter(Bid.status == 'active').first()

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

        timestamp = config.create_timestamp()

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

        new_bid = Bid(data)
        db.add(new_bid)
        db.commit()

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

        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_bid'

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

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

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

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