def setUp(self):
self.user_resource = UserResource()
self.user_resource_uri = self.user_resource.get_resource_uri()
self.client = TestApiClient()
super(BaseResourceTestCase, self).setUp()
def post(self):
response = {}
user_resource = UserResource(request.json)
try:
response['user_list'] = user_resource.get_or_create()
except Exception as error:
print error
pass
return self.json_response(data=response)
def post(self):
response = {}
user_resource = UserResource(request.json)
try:
response['user_list'] = user_resource.get_or_create()
except Exception as error:
print error
pass
return self.json_response(data=response)
def put(self, user=None, **kwargs):
response = {}
user_resource = UserResource(request.json, model=user)
if user_resource.is_valid():
try:
user_resource.update()
response["user"] = user_resource.to_serializable_dict()
except Exception as error:
print error
pass
return self.json_response(data=response)
def post(self):
response = {}
user_resource = UserResource(request.json)
if user_resource.is_valid():
try:
user_resource.add()
response["user"] = user_resource.to_serializable_dict()
except Exception as error:
print error
pass
return self.json_response(data=response)
def export_user(self, p_format):
try:
print("Exporting User")
from users.resources import UserResource
aresource = UserResource()
dataset = aresource.export()
tablename = 'User'
if p_format == "json":
self.save_to_file(a_filename=tablename+'.json',
a_content=dataset.json)
elif p_format == "csv":
self.save_to_file(a_filename=tablename+'.csv',
a_content=dataset.csv)
except Exception:
raise CommandError("User table could not be exported!")
def put(self, user=None, **kwargs):
response = {}
user_resource = UserResource(request.json, model=user)
if user_resource.is_valid():
try:
user_resource.update()
response['user'] = user_resource.to_serializable_dict()
except Exception as error:
print error
pass
return self.json_response(data=response)
def post(self):
response = {}
user_resource = UserResource(request.json)
if user_resource.is_valid():
try:
user_resource.add()
response['user'] = user_resource.to_serializable_dict()
except Exception as error:
print error
pass
return self.json_response(data=response)
def get(self, **kwargs):
resource = UserResource(request.json)
data ={'user_list': resource.filter_by()}
return self.json_response(data=data)
class BaseResourceTestCase(BaseTestCase, ResourceTestCase):
def setUp(self):
self.user_resource = UserResource()
self.user_resource_uri = self.user_resource.get_resource_uri()
self.client = TestApiClient()
super(BaseResourceTestCase, self).setUp()
def login(self, email=None, password=None):
if email is None:
email = self.user.email
if password is None:
password = self.password
url = "%slogin/" % self.user_resource_uri
data = {
"email": email,
"password": password,
}
if isinstance(self.client, Client):
data = json.dumps(data)
response = self.client.post(url, data=data,
content_type="application/json")
self.assertEqual(
response.status_code, 200,
"On '{}', expected status code: {}. Got {}: {}".format(
url,
200,
response.status_code,
response.content
)
)
def logout(self):
uri = "%slogout/" % self.user_resource_uri
data = {
}
response = self.client.get(uri, data=data,
content_type="application/json")
self.assertEqual(response.status_code, 200)
def create(self, resource, endpoint='', expected_code=201, **data):
return self.post(resource, expected_code=expected_code, **data)
def get(self, *args, **kwargs):
return self.call_api(method='get', *args, **kwargs)
def post(self, *args, **kwargs):
return self.call_api(method='post', *args, **kwargs)
def put(self, *args, **kwargs):
return self.call_api(method='put', *args, **kwargs)
def patch(self, *args, **kwargs):
return self.call_api(method='patch', *args, **kwargs)
def call_api(self, resource, obj=None, endpoint='', expected_code=200,
data=None, method='get', multipart=False):
if multipart:
self.client = Client()
self.login()
if isinstance(resource, basestring):
resource = api.canonical_resource_for(resource)
url = resource.get_resource_uri(obj)
if endpoint:
url = '{}{}/'.format(url, endpoint)
method = getattr(self.client, method.lower())
response = method(url, data=data)
self.assertEqual(
response.status_code, expected_code,
"On '{}', expected status code: {}. Got {}: {}".format(
url,
expected_code,
response.status_code,
response.content
)
)
try:
return json.loads(response.content)
except:
return response.content
def get(self, **kwargs):
resource = UserResource(request.json)
data = {'user_list': resource.filter_by()}
return self.json_response(data=data)
""" this document defines the project urls """
# django
from django.conf.urls import patterns, include, url
# resources
from users.resources import UserResource
# tastypie
from tastypie.api import Api
# api
from api.serializers import Serializer
api = Api(api_name='v1', serializer_class=Serializer)
api.register(UserResource())
urlpatterns = patterns(
'',
url(r'doc/', include('tastypie_swagger.urls',
namespace='tastypie_swagger')),
(r'', include(api.urls)),
)
