Change project structure and fix import issue

Author: lucrib

client/__init__.py

@@ -1,16 +1,14 @@
-from faker import Faker
 from flask import Flask
 from flask_restful import Api
 from flask_sqlalchemy import SQLAlchemy
 
-VERSION = '0.1.0'
-
-app = Flask('User')
+app = Flask(__name__)
 
 app.config['SECRET_KEY'] = 'LUCAS'
 app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///thisisatest.db'
 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
 
-api = Api(app)
 db = SQLAlchemy(app)
-faker = Faker()
+api = Api(app)
+
+import views

client/__main__.py

@@ -1,19 +1,15 @@
-from sqlalchemy import Column
-from sqlalchemy import Integer
-from sqlalchemy import String
-
 from client import db
 
 
 class User(db.Model):
     __tablename__ = 'user'
-    id = Column('id', Integer, primary_key=True)
-    name = Column('name', String)
-    age = Column('age', Integer)
-    birthdate = Column('birthdate', String)
-    email = Column('email', String)
-    username = Column('username', String)
-    password = Column('password', String)
+    id = db.Column('id', db.Integer, primary_key=True)
+    name = db.Column('name', db.String)
+    age = db.Column('age', db.Integer)
+    birthdate = db.Column('birthdate', db.String)
+    email = db.Column('email', db.String)
+    username = db.Column('username', db.String)
+    password = db.Column('password', db.String)
 
     def __init__(self, id=None, name=None, age=None, username=None, password=None, email=None, birthdate=None):
         self.id = id
@@ -29,6 +25,3 @@ def __repr__(self):
 
     def __str__(self):
         return u'User(id={}, username={}, name={}, email{})'.format(self.id, self.username, self.name, self.email)
-
-
-db.create_all()

client/endpoints.py

@@ -0,0 +1,7 @@
+from login import LoginResource
+from user import UserResource, UserListResource
+from client import api
+
+api.add_resource(LoginResource, '/login', endpoint='login')
+api.add_resource(UserResource, '/user/<int:id>', endpoint='user')
+api.add_resource(UserListResource, '/users', endpoint='users')

client/models.py

@@ -0,0 +1,31 @@
+# coding=utf-8
+
+import jwt
+from flask import request
+from flask_restful import Resource
+from client import app
+
+
+def valid_auth(auth):
+    if auth.username == 'username' and auth.password == 'password':
+        return True
+    return False
+
+
+class LoginResource(Resource):
+
+    def get(self):
+        auth = request.authorization
+        if not auth:
+            headers = {'WWW-Authenticate': "Basic realm='teste'"}
+            return {'message': 'Authentication must be provided.'}, 401, headers
+
+        if valid_auth(auth):
+            payload = dict(username=auth.username, password=auth.password)
+            jwt_data = jwt.encode(payload, app.config['SECRET_KEY'])
+            return {"jwt": jwt_data}, 200
+
+        return {'message': 'Invalid credentials.'}, 401
+
+
+

client/thisisatest.db

@@ -0,0 +1,37 @@
+import httplib as http
+
+from flask import request
+from flask_restful import Resource
+from flask_restful import fields, marshal_with
+
+from client import db
+from client.models import User
+
+user_fields = {
+    'id': fields.Integer,
+    'username': fields.String,
+    'email': fields.String,
+    'uri': fields.Url('user', absolute=True),
+}
+
+
+class UserListResource(Resource):
+    @marshal_with(user_fields)
+    def get(self):
+        users = User.query.all()
+        return users
+
+    @marshal_with(user_fields)
+    def post(self):
+        data = request.get_json()
+        user = User(**data)
+        db.session.add(user)
+        db.session.commit()
+        return user, http.CREATED
+
+
+class UserResource(Resource):
+    @marshal_with(user_fields)
+    def get(self, id):
+        user = User.query.filter_by(id=id).first()
+        return user

client/views/__init__.py

@@ -17,7 +17,9 @@ def step_impl(context, password):
 @when(u'I send a "{method}" request to the "{endpoint}" endpoint')
 def step_impl(context, method, endpoint):
     auth = HTTPBasicAuth(context.username, context.password)
-    context.response = request(method, context.base_uri + endpoint, auth=auth)
+    login_uri = context.base_uri + endpoint
+    print("Login URI ====>", login_uri)
+    context.response = request(method, login_uri, auth=auth)
 
 
 @then(u'I should receive a "{status_code}" status code')

client/views/login.py

@@ -1,12 +1,15 @@
+allure-behave==2.3.1b1
+allure-python-commons==2.3.1b1
 aniso8601==2.0.0
+attrs==17.4.0
 behave==1.2.5
-certifi==2017.11.5
+certifi==2018.1.18
 cffi==1.11.2
 chardet==3.0.4
 click==6.7
 coverage==4.4.2
 enum34==1.1.6
-Faker==0.8.8
+Faker==0.8.10
 Flask==0.12.2
 Flask-RESTful==0.3.6
 Flask-SQLAlchemy==2.3.2
@@ -17,6 +20,7 @@ Jinja2==2.10
 MarkupSafe==1.0
 parse==1.8.2
 parse-type==0.4.2
+pluggy==0.6.0
 pyasn1==0.4.2
 pycparser==2.18
 PyHamcrest==1.9.0
@@ -25,7 +29,7 @@ python-dateutil==2.6.1
 pytz==2017.3
 requests==2.18.4
 six==1.11.0
-SQLAlchemy==1.2.0
+SQLAlchemy==1.2.2
 text-unidecode==1.1
 urllib3==1.22
 Werkzeug==0.14.1

client/views/user.py

@@ -0,0 +1,4 @@
+from client import app
+
+if __name__ == '__main__':
+    app.run(debug=True, port=5000)