request_handler.py

import json
from http.server import BaseHTTPRequestHandler, HTTPServer
from views import get_all_posts, get_single_post, create_post, update_post, delete_post
from views import create_category, update_category, create_subscription, delete_subscription, update_subscription
from views import create_user, login_user, get_single_user, get_all_users, update_user, delete_user
from views import get_single_comment,get_all_comments,create_comment,delete_comment,update_comment
from views import get_single_posttags,get_all_posttags,create_posttag,delete_posttag,update_posttag
from views import get_all_categories,get_single_category,delete_category
from views import get_all_subscriptions, get_single_subscription, get_subscriptions_by_author
from views import get_all_tags,get_single_tag,update_tag,delete_tag,create_tag


class HandleRequests(BaseHTTPRequestHandler):
    """Handles the requests to this server"""

    def parse_url(self):
        """Parse the url into the resource and id"""
        path_params = self.path.split('/')
        resource = path_params[1]
        if '?' in resource:
            param = resource.split('?')[1]
            resource = resource.split('?')[0]
            pair = param.split('=')
            key = pair[0]
            value = pair[1]
            return (resource, key, value)
        else:
            id = None
            try:
                id = int(path_params[2])
            except (IndexError, ValueError):
                pass
            return (resource, id)

    def do_GET(self):
        """Handles GET request to the server"""
        self._set_headers(200)

        response = {}

        # If the path does not include a query parameter, continue with the original if block
        if '?' not in self.path:
            ( resource, id ) = self.parse_url()

            # It's an if..else statement
            if resource == "Comments":
                if id is not None:
                    response = get_single_comment(id)
                else:
                    response = get_all_comments()
                    
            if resource == "PostTags":
                if id is not None:
                    response= get_single_posttags(id)                    
                else:   
                    response= get_all_posttags()
            
            if resource == "Categories":
                if id is not None:
                    response = get_single_category(id)                
                else: 
                    response = get_all_categories()
            
            if resource == "Subscriptions":
                if id is not None:
                    response= get_single_subscription(id)
                else:
                    response= get_all_subscriptions()
            
            if resource == "Posts":
                if id is not None:
                    response= get_single_post(id)
                else:
                    response= get_all_posts()
                    
            if resource == "Users":
                if id is not None:
                    response = get_single_user(id)
                else:
                    response = get_all_users()

            if resource == "Tags":
                if id is not None:
                    response = get_single_tag(id)
                else:
                    response = get_all_tags()
        else:
            (resource, query, value) = self.parse_url()
            if resource == "Subscriptions" and query=="author_id":
                response = get_subscriptions_by_author(value)

        self.wfile.write(json.dumps(response).encode())
        
    def _set_headers(self, status):
        """Sets the status code, Content-Type and Access-Control-Allow-Origin
        headers on the response

        Args:
            status (number): the status code to return to the front end
        """
        self.send_response(status)
        self.send_header('Content-type', 'application/json')
        self.send_header('Access-Control-Allow-Origin', '*')
        self.end_headers()

    def do_OPTIONS(self):
        """Sets the OPTIONS headers
        """
        self.send_response(200)
        self.send_header('Access-Control-Allow-Origin', '*')
        self.send_header('Access-Control-Allow-Methods',
                            'GET, POST, PUT, DELETE')
        self.send_header('Access-Control-Allow-Headers',
                            'X-Requested-With, Content-Type, Accept')
        self.end_headers()

    def do_POST(self):
        """Make a post request to the server"""
        self._set_headers(201)
        content_len = int(self.headers.get('content-length', 0))
        post_body = json.loads(self.rfile.read(content_len))
        response = ''
        resource, _ = self.parse_url()

        if resource == 'login':
            response = login_user(post_body)
        if resource == 'register':
            response = create_user(post_body)
        if resource == 'Posts':
            response = create_post(post_body)
        if resource == 'Comments':
            response = create_comment(post_body)
        if resource == 'PostTags':
            response = create_posttag(post_body)
        if resource == 'Tags':
            response = create_tag(post_body)
        if resource == 'Categories':
            response = create_category(post_body)
        if resource == 'Subscriptions':
            response = create_subscription(post_body)

        self.wfile.write(json.dumps(response).encode())

    def do_PUT(self):
        """Handles PUT requests to the server"""
        content_len = int(self.headers.get('content-length', 0))
        post_body = self.rfile.read(content_len)
        post_body = json.loads(post_body)

        # Parse the URL
        (resource, id) = self.parse_url()
        # set default value of success
        success = False

        if resource == 'Posts': 
            success = update_post(id, post_body)
        if resource == 'Categories':
            success = update_category(id, post_body)
        if resource == 'Subscriptions':
            success = update_subscription(id, post_body)
        if resource == 'Comments':
            success = update_comment(id, post_body)
        if resource == 'PostTags':
            success = update_posttag(id, post_body)
        if resource == 'Tags':
            success = update_tag(id, post_body)
        if resource == 'Users':
            success = update_user(id, post_body)
            
        print(success)
        
        if success:
            self._set_headers(204)
        else:
            self._set_headers(404)
            
        self.wfile.write(json.dumps(success).encode())
        #self.wfile.write("".encode())


    def do_DELETE(self):
        self._set_headers(204)

        # Parse the URL
        (resource, id) = self.parse_url()

        # Delete a single animal from the list
        if resource == "Categories":
            delete_category(id)
        if resource == "Subscriptions":
            delete_subscription(id)
        if resource == "Comments":
            delete_comment(id)
        if resource == "Posts":
            delete_post(id) 
        if resource == "PostTags":
            delete_posttag(id)
        if resource == "Tags":
            delete_tag(id)
        if resource == 'Users':
            delete_user(id)
        
        
def main():
    """Starts the server on port 8088 using the HandleRequests class
    """
    host = ''
    port = 8088
    HTTPServer((host, port), HandleRequests).serve_forever()


if __name__ == "__main__":
    main()