def obj_create(self, bundle, **kwargs): """ Handles creating Events through the API. :param bundle: Bundle containing the information to create the Event. :type bundle: Tastypie Bundle object. :returns: HttpResponse. """ analyst = bundle.request.user.username title = bundle.data.get('title', None) description = bundle.data.get('description', None) event_type = bundle.data.get('event_type', None) source = bundle.data.get('source', None) method = bundle.data.get('method', None) reference = bundle.data.get('reference', None) date = bundle.data.get('date', None) bucket_list = bundle.data.get('bucket_list', None) ticket = bundle.data.get('ticket', None) campaign = bundle.data.get('campaign', None) campaign_confidence = bundle.data.get('campaign_confidence', None) content = {'return_code': 0, 'type': 'Event'} if not title or not event_type or not source or not description: content[ 'message'] = 'Must provide a title, event_type, source, and description.' self.cripts_response(content) if event_type not in EventTypes.values(): content['message'] = 'Not a valid Event Type.' self.cripts_response(content) result = add_new_event( title, description, event_type, source, method, reference, date, analyst, bucket_list, ticket, ) if result.get('message'): content['message'] = result.get('message') content['id'] = result.get('id', '') if result.get('id'): url = reverse('api_dispatch_detail', kwargs={ 'resource_name': 'events', 'api_name': 'v1', 'pk': result.get('id') }) content['url'] = url if result['success']: content['return_code'] = 0 else: content['return_code'] = 1 self.cripts_response(content)
def obj_create(self, bundle, **kwargs): """ Handles creating Events through the API. :param bundle: Bundle containing the information to create the Event. :type bundle: Tastypie Bundle object. :returns: HttpResponse. """ analyst = bundle.request.user.username title = bundle.data.get('title', None) description = bundle.data.get('description', None) event_type = bundle.data.get('event_type', None) source = bundle.data.get('source', None) method = bundle.data.get('method', None) reference = bundle.data.get('reference', None) date = bundle.data.get('date', None) bucket_list = bundle.data.get('bucket_list', None) ticket = bundle.data.get('ticket', None) campaign = bundle.data.get('campaign', None) campaign_confidence = bundle.data.get('campaign_confidence', None) content = {'return_code': 0, 'type': 'Event'} if not title or not event_type or not source or not description: content['message'] = 'Must provide a title, event_type, source, and description.' self.cripts_response(content) if event_type not in EventTypes.values(): content['message'] = 'Not a valid Event Type.' self.cripts_response(content) result = add_new_event(title, description, event_type, source, method, reference, date, analyst, bucket_list, ticket, ) if result.get('message'): content['message'] = result.get('message') content['id'] = result.get('id', '') if result.get('id'): url = reverse('api_dispatch_detail', kwargs={'resource_name': 'events', 'api_name': 'v1', 'pk': result.get('id')}) content['url'] = url if result['success']: content['return_code'] = 0 else: content['return_code'] = 1 self.cripts_response(content)
def __init__(self, username, *args, **kwargs): super(EventForm, self).__init__(*args, **kwargs) self.fields["source"].choices = [(c.name, c.name) for c in get_source_names(True, True, username)] self.fields["source"].initial = get_user_organization(username) self.fields["event_type"].choices = [(c, c) for c in EventTypes.values(sort=True)] self.fields["relationship_type"].choices = relationship_choices self.fields["relationship_type"].initial = RelationshipTypes.RELATED_TO add_bucketlist_to_form(self) add_ticket_to_form(self)
def set_event_type(self, event_type): """ Set the Event Type. :param event_type: The event type to set (must exist in DB). :type event_type: str """ if event_type in EventTypes.values(): self.event_type = event_type
def __init__(self, username, *args, **kwargs): super(EventForm, self).__init__(*args, **kwargs) self.fields['source'].choices = [ (c.name, c.name) for c in get_source_names(True, True, username) ] self.fields['source'].initial = get_user_organization(username) self.fields['event_type'].choices = [ (c, c) for c in EventTypes.values(sort=True) ] self.fields['relationship_type'].choices = relationship_choices self.fields['relationship_type'].initial = RelationshipTypes.RELATED_TO add_bucketlist_to_form(self) add_ticket_to_form(self)
def get_event_type_dropdown(request): """ Get a list of available event types. :param request: Django request object (Required) :type request: :class:`django.http.HttpRequest` :returns: :class:`django.http.HttpResponse` """ if request.is_ajax(): e_types = EventTypes.values(sort=True) result = {'types': e_types} return HttpResponse(json.dumps(result), content_type="application/json") else: error = "Expected AJAX" return render_to_response("error.html", {"error": error}, RequestContext(request))