def test_ignore_endpoints(self):
self.app.config['SERVER_NAME'] = 'www.example.com'
self.app.config['SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS'] = True
self.app.config['SITEMAP_IGNORE_ENDPOINTS'] = ['first', 'user']
sitemap = Sitemap(app=self.app)
now = datetime.now().isoformat()
@self.app.route('/')
def index():
pass
@self.app.route('/first')
def first():
pass
@self.app.route('/second')
def second():
pass
@self.app.route('/<username>')
def user(username):
pass
@sitemap.register_generator
def user():
yield 'user', {'username': '******'}
yield 'user', {'username': '******'}, now
results = [result['loc'] for result in sitemap._generate_all_urls()]
assert 'http://www.example.com/' in results
assert 'http://www.example.com/first' not in results
assert 'http://www.example.com/second' in results
assert 'http://www.example.com/third' not in results
assert 'http://www.example.com/fourth' not in results
def test_url_generator(self):
self.app.config['SERVER_NAME'] = 'www.example.com'
sitemap = Sitemap(app=self.app)
now = datetime.now().isoformat()
@self.app.route('/')
def index():
pass
@self.app.route('/<username>')
def user(username):
pass
@sitemap.register_generator
def user():
yield 'http://www.example.com/first'
yield {'username': '******'}
yield 'user', {'username': '******'}
yield 'user', {'username': '******'}, now
results = sitemap._generate_all_urls()
assert next(results)['loc'] == 'http://www.example.com/first'
assert next(results)['loc'] == 'http://www.example.com/second'
assert next(results)['loc'] == 'http://www.example.com/third'
assert next(results)['loc'] == 'http://www.example.com/fourth'
with open(os.path.join(
os.path.dirname(__file__), 'data', 'sitemap.xml'), 'r') as f:
out = f.read().format(now=now).strip()
assert out == sitemap.sitemap()
def test_ignore_endpoints(self):
self.app.config['SERVER_NAME'] = 'www.example.com'
self.app.config['SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS'] = True
self.app.config['SITEMAP_IGNORE_ENDPOINTS'] = ['first', 'user']
sitemap = Sitemap(app=self.app)
now = datetime.now().isoformat()
@self.app.route('/')
def index():
pass
@self.app.route('/first')
def first():
pass
@self.app.route('/second')
def second():
pass
@self.app.route('/<username>')
def user(username):
pass
@sitemap.register_generator
def user():
yield 'user', {'username': '******'}
yield 'user', {'username': '******'}, now
results = [result['loc'] for result in sitemap._generate_all_urls()]
assert 'http://www.example.com/' in results
assert 'http://www.example.com/first' not in results
assert 'http://www.example.com/second' in results
assert 'http://www.example.com/third' not in results
assert 'http://www.example.com/fourth' not in results
def test_url_generator(self):
self.app.config['SERVER_NAME'] = 'www.example.com'
sitemap = Sitemap(app=self.app)
now = datetime.now().isoformat()
@self.app.route('/')
def index():
pass
@self.app.route('/<username>')
def user(username):
pass
@sitemap.register_generator
def user():
yield 'http://www.example.com/first'
yield {'username': '******'}
yield 'user', {'username': '******'}
yield 'user', {'username': '******'}, now
results = sitemap._generate_all_urls()
assert next(results)['loc'] == 'http://www.example.com/first'
assert next(results)['loc'] == 'http://www.example.com/second'
assert next(results)['loc'] == 'http://www.example.com/third'
assert next(results)['loc'] == 'http://www.example.com/fourth'
with open(
os.path.join(os.path.dirname(__file__), 'data', 'sitemap.xml'),
'r') as f:
out = f.read().format(now=now).strip()
assert out == sitemap.sitemap()
def test_signals(self):
now = datetime.now().isoformat()
cache = {}
@sitemap_page_needed.connect
def create_page(app, page, urlset):
cache[page] = sitemap.render_page(urlset=urlset)
def load_page(fn):
from functools import wraps
@wraps(fn)
def loader(*args, **kwargs):
page = kwargs.get('page')
data = cache.get(page)
return data if data else fn(*args, **kwargs)
return loader
self.app.config['SERVER_NAME'] = 'www.example.com'
self.app.config['SITEMAP_GZIP'] = True
self.app.config['SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS'] = True
self.app.config['SITEMAP_MAX_URL_COUNT'] = 10
self.app.config['SITEMAP_VIEW_DECORATORS'] = [load_page]
sitemap = Sitemap(app=self.app)
@self.app.route('/')
def index():
pass
@self.app.route('/first')
def first():
pass
@self.app.route('/second')
def second():
pass
@self.app.route('/<username>')
def user(username):
pass
@sitemap.register_generator
def user():
for number in range(20):
yield 'user', {'username': '******'.format(number)}
with self.app.test_client() as c:
assert 200 == c.get('/sitemap.xml').status_code
assert len(cache) == 3
for page in range(len(cache)):
assert sitemap.gzip_response(cache[page + 1]).data == c.get(
'/sitemap{0}.xml'.format(page + 1)).data
def test_signals(self):
now = datetime.now().isoformat()
cache = {}
@sitemap_page_needed.connect
def create_page(app, page, urlset):
cache[page] = sitemap.render_page(urlset=urlset)
def load_page(fn):
from functools import wraps
@wraps(fn)
def loader(*args, **kwargs):
page = kwargs.get('page')
data = cache.get(page)
return data if data else fn(*args, **kwargs)
return loader
self.app.config['SERVER_NAME'] = 'www.example.com'
self.app.config['SITEMAP_GZIP'] = True
self.app.config['SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS'] = True
self.app.config['SITEMAP_MAX_URL_COUNT'] = 10
self.app.config['SITEMAP_VIEW_DECORATORS'] = [load_page]
sitemap = Sitemap(app=self.app)
@self.app.route('/')
def index():
pass
@self.app.route('/first')
def first():
pass
@self.app.route('/second')
def second():
pass
@self.app.route('/<username>')
def user(username):
pass
@sitemap.register_generator
def user():
for number in range(20):
yield 'user', {'username': '******'.format(number)}
with self.app.test_client() as c:
assert 200 == c.get('/sitemap.xml').status_code
assert len(cache) == 3
for page in range(len(cache)):
assert sitemap.gzip_response(cache[page+1]).data == c.get(
'/sitemap{0}.xml'.format(page+1)).data
def test_decorators_order(self):
def first(dummy):
return lambda *args, **kwargs: 'first'
def second(dummy):
return lambda *args, **kwargs: 'second'
def third(dummy):
return lambda *args, **kwargs: 'third'
self.app.config['SITEMAP_VIEW_DECORATORS'] = [
first, second, 'tests.helpers.dummy_decorator'
]
sitemap = Sitemap(app=self.app)
assert first in sitemap.decorators
assert second in sitemap.decorators
with self.app.test_client() as c:
assert b'dummy' == c.get('/sitemap.xml').data
sitemap.decorators.append(third)
with self.app.test_client() as c:
assert b'third' == c.get('/sitemap.xml').data
def test_pagination(self):
self.app.config['SERVER_NAME'] = 'www.example.com'
self.app.config['SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS'] = True
self.app.config['SITEMAP_MAX_URL_COUNT'] = 10
sitemap = Sitemap(app=self.app)
now = datetime.now().isoformat()
@self.app.route('/')
def index():
pass
@self.app.route('/first')
def first():
pass
@self.app.route('/second')
def second():
pass
@self.app.route('/<username>')
def user(username):
pass
@sitemap.register_generator
def user():
for number in range(20):
yield 'user', {'username': '******'.format(number)}
with self.app.test_client() as c:
assert b('sitemapindex') in c.get('/sitemap.xml').data
assert len(c.get('/sitemap1.xml').data) > 0
def init_app(self, app):
"""Flask application initialization."""
self.init_config(app)
self.cached_pages_set_key = 'sitemap_page_keys' # Keep track of cached pages
app.extensions['weko-sitemap'] = self
app.config['SITEMAP_VIEW_DECORATORS'] = [self.load_page]
ext = Sitemap(app=app)
ext.register_generator(self._generate_all_item_urls)
ext.blueprint = Blueprint('flask_sitemap',
'flask_sitemap', template_folder='templates')
ext.blueprint.add_url_rule( # Return cached sitemap or update it
app.config.get('SITEMAP_ENDPOINT_URL'),
'sitemap',
ext._decorate(self.sitemap),
)
ext.blueprint.add_url_rule(
app.config.get('SITEMAP_ENDPOINT_PAGE_URL'),
'page',
self.page # Do not decorate
)
app.register_blueprint(
ext.blueprint,
url_prefix=app.config.get('SITEMAP_BLUEPRINT_URL_PREFIX')
)
def sitemap():
app_instance.config["SERVER_NAME"] = "atom.shortcircuits.io"
sitemap = Sitemap(app=app_instance)
@app_instance.route('/package/<name>')
def package(name):
pass
@app_instance.route("/")
def index():
pass
@app_instance.route('/search')
def search():
pass
@sitemap.register_generator
def package_urls():
import datetime
from app.packages.models import Package, Downloads
yield 'index', {}, datetime.datetime.today().date().isoformat(
), "daily", 1
yield 'search', {}, datetime.datetime.today().date().isoformat(
), "daily", 0.5
package_list = Package.query.all()
for item in package_list:
latest = item.downloads.order_by(Downloads.date.desc()).first()
time = latest.date.date().isoformat() if latest is not None else ""
yield 'package', {'name': item.name}, time, "daily", 0.5
out = sitemap.sitemap()
if not os.path.exists(SITEMAP_DIR):
os.makedirs(SITEMAP_DIR)
f = open(os.path.join(SITEMAP_DIR, SITEMAP_NAME), 'w')
f.write(out)
f.close()
def sitemap():
app_instance.config["SERVER_NAME"] = "atom.shortcircuits.io"
sitemap = Sitemap(app=app_instance)
@app_instance.route('/package/<name>')
def package(name):
pass
@app_instance.route("/")
def index():
pass
@app_instance.route('/search')
def search():
pass
@sitemap.register_generator
def package_urls():
import datetime
from app.packages.models import Package, Downloads
yield 'index', {}, datetime.datetime.today().date().isoformat(), "daily", 1
yield 'search', {}, datetime.datetime.today().date().isoformat(), "daily", 0.5
package_list = Package.query.all()
for item in package_list:
latest = item.downloads.order_by(Downloads.date.desc()).first()
time = latest.date.date().isoformat() if latest is not None else ""
yield 'package', {'name': item.name}, time, "daily", 0.5
out = sitemap.sitemap()
if not os.path.exists(SITEMAP_DIR):
os.makedirs(SITEMAP_DIR)
f = open(os.path.join(SITEMAP_DIR, SITEMAP_NAME), 'w')
f.write(out)
f.close()
def register(self, app, options, first_registration=False):
sitemap = Sitemap(app=app)
@sitemap.register_generator
def index():
# (route, options, lastmod, changefreq, priority)
for route in self.STATIC_ROUTES:
yield (route, {}, None, 'monthly', '0.5')
brigades = get_brigades()
for brigade in json.loads(brigades):
last_updated = datetime.fromtimestamp(
brigade['properties']['last_updated'])
for route in self.PER_BRIGADE_ROUTES:
yield (route, {
'brigadeid': brigade['id']
}, last_updated.strftime("%Y-%m-%d"), 'weekly', '1.0')
def init_app(app):
env = app.config.ENV
db.init_app(app)
if env == "development" or env == "testing":
development(app)
production(app)
sitemap = Sitemap(app)
@sitemap.register_generator
def _():
from app.models.content import Content
yield "home.index", {}
for content in Content.query.all():
yield "contents.show", {"content_slug": content.slug}
def register(self, app, options, first_registration=False):
sitemap = Sitemap(app=app)
@app.route('/robots.txt')
def static_from_root():
return send_from_directory(app.static_folder, request.path[1:])
@sitemap.register_generator
def index():
# (route, options, lastmod, changefreq, priority)
for route in self.STATIC_ROUTES:
yield (route, {}, None, 'monthly', '0.5')
brigades = get_brigades()
for brigade in brigades:
last_updated = datetime.fromtimestamp(
brigade['properties']['last_updated'])
for route in self.PER_BRIGADE_ROUTES:
yield (route, {
'brigadeid': brigade['id']
}, last_updated.strftime("%Y-%m-%d"), 'weekly', '1.0')
def test_creation_init(self):
assert 'sitemap' not in self.app.extensions
r = Sitemap()
r.init_app(app=self.app)
assert isinstance(self.app.extensions['sitemap'], Sitemap)
def test_creation_old_flask(self):
# Simulate old Flask (pre 0.9)
del self.app.extensions
Sitemap(app=self.app)
assert isinstance(self.app.extensions['sitemap'], Sitemap)
def test_creation(self):
assert 'sitemap' not in self.app.extensions
Sitemap(app=self.app)
assert isinstance(self.app.extensions['sitemap'], Sitemap)
def test_pagination(self):
self.app.config['SERVER_NAME'] = 'www.example.com'
self.app.config['SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS'] = True
self.app.config['SITEMAP_MAX_URL_COUNT'] = 10
sitemap = Sitemap(app=self.app)
now = datetime.now().isoformat()
@self.app.route('/')
def index():
pass
@self.app.route('/first')
def first():
pass
@self.app.route('/second')
def second():
pass
@self.app.route('/<username>')
def user(username):
pass
@sitemap.register_generator
def user():
for number in range(20):
yield 'user', {'username': '******'.format(number)}
directory = mkdtemp()
runner = CliRunner()
try:
result = runner.invoke(
self.app.cli,
['sitemap', '-o', directory, '-v'],
obj=ScriptInfo(create_app=lambda _: self.app),
)
assert 'sitemap1.xml\nsitemap2.xml\nsitemap3.xml\nsitemap.xml' \
in result.output
# assert result.exit_code == 0
with self.app.test_client() as c:
data = c.get('/sitemap.xml').data
data1 = c.get('/sitemap1.xml').data
assert b'sitemapindex' in data
assert len(data1) > 0
with open(os.path.join(directory, 'sitemap.xml'), 'rb') as f:
assert f.read() == data
with open(os.path.join(directory, 'sitemap1.xml'), 'rb') as f:
assert f.read() == data1
finally:
shutil.rmtree(directory)
directory = mkdtemp()
manager = Manager(self.app)
manager.add_command('sitemap', script.Sitemap())
try:
manager.handle('manage.py', ['sitemap', '-o', directory])
with self.app.test_client() as c:
data = c.get('/sitemap.xml').data
data1 = c.get('/sitemap1.xml').data
assert b'sitemapindex' in data
assert len(data1) > 0
with open(os.path.join(directory, 'sitemap.xml'), 'rb') as f:
assert f.read() == data
with open(os.path.join(directory, 'sitemap1.xml'), 'rb') as f:
assert f.read() == data1
finally:
shutil.rmtree(directory)
extra_fields = dict(
category_field=config.get('category_field'), desc_field=config.get('desc_field'),
link_field=config.get('link_field'), priority_field=config.get('priority_field'),
order_field=config.get('order_field'))
category_field, desc_field, link_field, priority_field, order_field = extra_fields.values()
# Workspace configuration
workspaces = config.get('workspaces')
workspaces_by_page = {w['page']: w['id'] for w in workspaces}
all_workspaces = ','.join(str(w['id']) for w in workspaces)
# Flask configuration
app = application = Flask(__name__)
app.secret_key = config.get('secret_key')
app.config.from_mapping(config.get('flask_config', {}))
cache, sitemap = Cache(app), Sitemap(app)
# Logging configuration
logger = logging.getLogger(__name__)
logger.addHandler(logging.StreamHandler())
logger.setLevel(config.get('log_level') or 'INFO')
def get_startups(workspace=None, category=None, search=None, startup_id=None):
"""
Get startups from Motherbase with activities, entity types, linkedin and twitter accounts
"""
results = {}
cache_file = os.path.join(cache_dir, f'.startups_{workspace or "all"}_{category or "all"}.json')
try:
if search:
def test_creation_init(self):
assert 'sitemap' not in self.app.extensions
r = Sitemap()
r.init_app(app=self.app)
assert isinstance(self.app.extensions['sitemap'], Sitemap)
def test_empty_generator(self):
Sitemap(app=self.app)
with self.app.test_client() as c:
assert b'loc' not in c.get('/sitemap.xml').data
def test_double_creation(self):
Sitemap(app=self.app)
self.assertRaises(RuntimeError, Sitemap, app=self.app)
from flask import Flask
from flask_bootstrap import Bootstrap
from flask_sitemap import Sitemap
from flask_wtf.csrf import CSRFProtect
import os
bootstrap = Bootstrap()
sitemap = Sitemap()
csrf = CSRFProtect()
if os.getenv('SERVER_SOFTWARE', '').startswith('Google App Engine/'):
app = Flask(__name__,
root_path='blog',
template_folder='templates/production',
instance_relative_config=True)
app.config.from_object('blog.settings.Production')
app.jinja_env.globals['DEV'] = False
else:
app = Flask(__name__, root_path='blog', instance_relative_config=True)
app.config.from_object('blog.settings.Testing')
app.jinja_env.globals['DEV'] = True
bootstrap.init_app(app)
from flask import Flask
from flask_bootstrap import Bootstrap
from flask_sitemap import Sitemap
from flask_wtf.csrf import CSRFProtect
import os
bootstrap = Bootstrap()
sitemap = Sitemap()
csrf = CSRFProtect()
if os.getenv('SERVER_SOFTWARE', '').startswith('Google App Engine/'):
app = Flask(__name__, root_path='blog',
template_folder= 'templates/production',
instance_relative_config = True)
from flask_sitemap import Sitemap
from syspath import git_root
dotenv.load_dotenv(os.path.join(git_root.path, '.env'))
from app.routes import handlers, sitemap_urls
app = Flask(__name__)
app.debug = os.environ['DEBUG'] == 'true'
if os.environ.get('SERVER_NAME', ''): # pragma: no cover
app.config['SERVER_NAME'] = os.environ['SERVER_NAME']
app.config['SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS'] = True
app.config['SITEMAP_URL_SCHEME'] = 'https'
assets = Environment(app)
ext = Sitemap(app=app)
ext.register_generator(sitemap_urls)
js = Bundle(
'js/jquery.js',
'js/bootstrap.bundle.js',
'js/pdfobject.js',
'js/global.js',
filters='rjsmin', output='gen/bundle.min.js'
)
assets.register('js_all', js)
css = Bundle(
'css/normalize.css',
'css/bootstrap.css',
'css/syntax.css',
def __init__(self, import_name):
super().__init__(import_name,
static_url_path='',
static_folder=os.path.join(os.getcwd(), 'web', 'static'),
template_folder=os.path.join(os.getcwd(), 'web', 'templates', 'public'))
self.wsgi_app = ProxyFix(self.wsgi_app)
self.data_store_client = datastore.Client()
self.client_secret = CLIENT_SECRET
self.firebase_admin_secret = FIREBASE_ADMIN_SECRET
self.firebase_admin_credentials = FIREBASE_ADMIN_CREDENTIALS
firebase_admin.initialize_app(self.firebase_admin_credentials)
# database
self.config["SQLALCHEMY_DATABASE_URI"] = SQLALCHEMY_DATABASE_URI
# self.config["SQLALCHEMY_ECHO"] = SQLALCHEMY_ECHO
self.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = SQLALCHEMY_TRACK_MODIFICATIONS
self.config["SQLALCHEMY_POOL_TIMEOUT"] = SQLALCHEMY_POOL_TIMEOUT
self.config["SQLALCHEMY_MAX_OVERFLOW"] = SQLALCHEMY_MAX_OVERFLOW
self.config['SWAGGER_UI_DOC_EXPANSION'] = SWAGGER_UI_DOC_EXPANSION
self.config['RESTPLUS_VALIDATE'] = RESTPLUS_VALIDATE
self.config['RESTPLUS_MASK_SWAGGER'] = RESTPLUS_MASK_SWAGGER
self.config['ERROR_404_HELP'] = ERROR_404_HELP
self.db = db
self.db.init_app(self)
self.app_context().push()
with self.app_context():
self.db.create_all() # Create database tables for our data models
self.flow = None
self.session = dict()
self.cache = Cache(app=self, config={'CACHE_TYPE': 'simple'})
self.config['SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS'] = False # true for listing every route available
self.flask_sitemap = Sitemap(app=self)
# keep a sitemap only for the pages that does not require authentication (may add special cases later)
self.flask_sitemap.register_generator(self.root_sitemap)
self.add_url_rule('/', view_func=self.landing, methods=['GET'])
self.add_url_rule('/profile', view_func=self.profile, methods=['GET'])
self.add_url_rule('/dashboard', view_func=self.dashboard, methods=['GET'])
self.add_url_rule('/categories', view_func=self.categories, methods=['GET', 'POST'])
self.add_url_rule('/add_category', view_func=self.add_category, methods=['GET', 'POST'])
self.add_url_rule('/delete_category/<int:category_id>', view_func=self.delete_category, methods=['GET'])
self.add_url_rule('/my_projects', view_func=self.my_projects, methods=['GET'])
self.add_url_rule('/add_project', view_func=self.add_project, methods=['GET', 'POST'])
self.add_url_rule('/see_project/<int:project_id>', view_func=self.see_project, methods=['GET'])
self.add_url_rule('/delete_project/<int:project_id>', view_func=self.delete_project, methods=['GET'])
self.add_url_rule('/my_jobs', view_func=self.my_jobs, methods=['GET'])
self.add_url_rule('/see_job/<int:job_id>', view_func=self.see_job, methods=['GET'])
self.add_url_rule('/delete_job/<int:job_id>', view_func=self.delete_job, methods=['GET'])
self.add_url_rule('/add_job', view_func=self.add_job, methods=['GET', 'POST'])
self.add_url_rule('/marketplace', view_func=self.marketplace, methods=['GET'])
self.add_url_rule('/product/<int:product_id>', view_func=self.product, methods=['GET'])
self.add_url_rule('/logout', view_func=self.logout, methods=['GET'])
self.add_url_rule('/login', view_func=self.login, methods=['GET'])
# APIs endpoints
blueprint = Blueprint('api', __name__, url_prefix='/api')
api.init_app(blueprint)
api.add_namespace(categories_namespace)
api.add_namespace(jobs_namespace)
api.add_namespace(attachments_namespace)
api.add_namespace(biddings_namespace)
api.add_namespace(projects_namespace)
api.add_namespace(projects_assets_namespace)
api.add_namespace(marketplace_namespace)
api.add_namespace(assets_namespace)
self.register_blueprint(blueprint)
self.register_error_handler(500, self.server_error)
self.register_error_handler(404, self.not_found)
class App(Flask):
def __init__(self, import_name):
super().__init__(import_name,
static_url_path='',
static_folder=os.path.join(os.getcwd(), 'web', 'static'),
template_folder=os.path.join(os.getcwd(), 'web', 'templates', 'public'))
self.wsgi_app = ProxyFix(self.wsgi_app)
self.data_store_client = datastore.Client()
self.client_secret = CLIENT_SECRET
self.firebase_admin_secret = FIREBASE_ADMIN_SECRET
self.firebase_admin_credentials = FIREBASE_ADMIN_CREDENTIALS
firebase_admin.initialize_app(self.firebase_admin_credentials)
# database
self.config["SQLALCHEMY_DATABASE_URI"] = SQLALCHEMY_DATABASE_URI
# self.config["SQLALCHEMY_ECHO"] = SQLALCHEMY_ECHO
self.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = SQLALCHEMY_TRACK_MODIFICATIONS
self.config["SQLALCHEMY_POOL_TIMEOUT"] = SQLALCHEMY_POOL_TIMEOUT
self.config["SQLALCHEMY_MAX_OVERFLOW"] = SQLALCHEMY_MAX_OVERFLOW
self.config['SWAGGER_UI_DOC_EXPANSION'] = SWAGGER_UI_DOC_EXPANSION
self.config['RESTPLUS_VALIDATE'] = RESTPLUS_VALIDATE
self.config['RESTPLUS_MASK_SWAGGER'] = RESTPLUS_MASK_SWAGGER
self.config['ERROR_404_HELP'] = ERROR_404_HELP
self.db = db
self.db.init_app(self)
self.app_context().push()
with self.app_context():
self.db.create_all() # Create database tables for our data models
self.flow = None
self.session = dict()
self.cache = Cache(app=self, config={'CACHE_TYPE': 'simple'})
self.config['SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS'] = False # true for listing every route available
self.flask_sitemap = Sitemap(app=self)
# keep a sitemap only for the pages that does not require authentication (may add special cases later)
self.flask_sitemap.register_generator(self.root_sitemap)
self.add_url_rule('/', view_func=self.landing, methods=['GET'])
self.add_url_rule('/profile', view_func=self.profile, methods=['GET'])
self.add_url_rule('/dashboard', view_func=self.dashboard, methods=['GET'])
self.add_url_rule('/categories', view_func=self.categories, methods=['GET', 'POST'])
self.add_url_rule('/add_category', view_func=self.add_category, methods=['GET', 'POST'])
self.add_url_rule('/delete_category/<int:category_id>', view_func=self.delete_category, methods=['GET'])
self.add_url_rule('/my_projects', view_func=self.my_projects, methods=['GET'])
self.add_url_rule('/add_project', view_func=self.add_project, methods=['GET', 'POST'])
self.add_url_rule('/see_project/<int:project_id>', view_func=self.see_project, methods=['GET'])
self.add_url_rule('/delete_project/<int:project_id>', view_func=self.delete_project, methods=['GET'])
self.add_url_rule('/my_jobs', view_func=self.my_jobs, methods=['GET'])
self.add_url_rule('/see_job/<int:job_id>', view_func=self.see_job, methods=['GET'])
self.add_url_rule('/delete_job/<int:job_id>', view_func=self.delete_job, methods=['GET'])
self.add_url_rule('/add_job', view_func=self.add_job, methods=['GET', 'POST'])
self.add_url_rule('/marketplace', view_func=self.marketplace, methods=['GET'])
self.add_url_rule('/product/<int:product_id>', view_func=self.product, methods=['GET'])
self.add_url_rule('/logout', view_func=self.logout, methods=['GET'])
self.add_url_rule('/login', view_func=self.login, methods=['GET'])
# APIs endpoints
blueprint = Blueprint('api', __name__, url_prefix='/api')
api.init_app(blueprint)
api.add_namespace(categories_namespace)
api.add_namespace(jobs_namespace)
api.add_namespace(attachments_namespace)
api.add_namespace(biddings_namespace)
api.add_namespace(projects_namespace)
api.add_namespace(projects_assets_namespace)
api.add_namespace(marketplace_namespace)
api.add_namespace(assets_namespace)
self.register_blueprint(blueprint)
self.register_error_handler(500, self.server_error)
self.register_error_handler(404, self.not_found)
def store_time(self, email, dt):
entity = datastore.Entity(key=self.data_store_client.key('User', email, 'visit'))
entity.update({
'timestamp': dt
})
self.data_store_client.put(entity)
def fetch_times(self, email, limit):
ancestor = self.data_store_client.key('User', email)
query = self.data_store_client.query(kind='visit', ancestor=ancestor)
query.order = ['-timestamp']
times = query.fetch(limit=limit)
return times
def landing(self):
return render_template('index.html')
@staticmethod
def root_sitemap():
# Not needed if you set SITEMAP_INCLUDE_RULES_WITHOUT_PARAMS=True
yield 'landing', {}
def get_claim(self):
# Verify Firebase auth.
id_token = request.cookies.get("token")
error_message = None
claims = None
times = None
if id_token:
try:
# Verify the token against the Firebase Auth API. This example
# verifies the token on each page load. For improved performance,
# some applications may wish to cache results in an encrypted
# session store (see for instance
claims = google.oauth2.id_token.verify_firebase_token(id_token, google_requests.Request())
if 'current_user' not in self.session:
self.session['current_user'] = dict()
sign_in_provider = claims['firebase']['sign_in_provider']
if sign_in_provider == 'phone':
self.store_time(claims['phone_number'], datetime.datetime.now())
times = self.fetch_times(claims['phone_number'], 10)
userInfo = auth.get_user_by_phone_number(claims['phone_number'])
else:
self.store_time(claims['email'], datetime.datetime.now())
times = self.fetch_times(claims['email'], 10)
userInfo = auth.get_user_by_email(claims['email'])
self.session['claims'] = claims
self.session['current_user']['uid'] = userInfo.uid
self.session['current_user']['phone_number'] = userInfo.phone_number
self.session['current_user']['email'] = userInfo.email
self.session['current_user']['provider_id'] = userInfo.provider_id
self.session['current_user']['disabled'] = userInfo.disabled
self.session['current_user']['tokens_valid_after_timestamp'] = userInfo.tokens_valid_after_timestamp
self.session['current_user']['creation_timestamp'] = userInfo.user_metadata.creation_timestamp
self.session['current_user']['last_sign_in_timestamp'] = userInfo.user_metadata.last_sign_in_timestamp
self.session['current_user']['provider_data'] = userInfo.provider_data
self.session['current_user']['custom_claims'] = userInfo.custom_claims
self.session['current_user']['tenant_id'] = userInfo.tenant_id
except ValueError as exc:
# This will be raised if the token is expired or any other
# verification checks fail.
error_message = str(exc)
return claims, times, error_message
def login_required(func):
@wraps(func)
def function_wrapper(*args):
if 'claims' in args[0].session:
claims = args[0].session['claims']
if claims is not None and datetime.datetime.now().timestamp() < claims['exp']:
claims, times, error_message = args[0].get_claim()
if error_message is None:
return func(*args)
args[0].session = dict()
return args[0].unauthorized_handler()
else:
return args[0].unauthorized_handler()
return function_wrapper
def login(self):
claims, times, error_message = self.get_claim()
if error_message is None and claims is not None:
return redirect(url_for('dashboard'), code=302)
return render_template('login.html')
@login_required
def logout(self):
auth.revoke_refresh_tokens(self.session['current_user']['uid'])
self.session = dict()
return redirect(url_for('login'))
@login_required
def profile(self):
project_list = get_projects_by_email(self.session['claims']['email'])
projects_count = len(project_list)
projects_in_progress = projects_count
job_list = get_jobs(page=1, per_page=50, category_id=None,
user_email=self.session['claims']['email'], freelancer_flag=False).items
jobs_count = len(job_list)
user_data = {
'projects_count': projects_count,
'projects_in_progress': projects_in_progress,
'jobs_count': jobs_count
}
return render_template('profile.html', session=self.session, user_data=user_data)
@login_required
def dashboard(self):
project_list = get_projects_by_email(self.session['claims']['email'])
projects_count = len(project_list)
projects_past_deadline_count = len(list(filter(lambda p: p.deadline < datetime.datetime.now(), project_list)))
projects_in_progress = projects_count
bidding_list = get_biddings_by_email(self.session['claims']['email'])
biddings_count = len(bidding_list)
job_list = get_jobs(page=1, per_page=50, category_id=None,
user_email=self.session['claims']['email'], freelancer_flag=False).items
jobs_count = len(job_list)
user_data = {
'projects_count': projects_count,
'projects_past_deadline': projects_past_deadline_count,
'projects_in_progress': projects_in_progress,
'biddings_count': biddings_count,
'jobs_count': jobs_count
}
return render_template('dashboard.html', session=self.session, user_data=user_data)
@login_required
def categories(self):
category_list = get_categories()
return render_template('categories.html', session=self.session, category_list=category_list)
@staticmethod
def delete_category(category_id):
delete_category(category_id)
return redirect(url_for('categories'))
@login_required
def add_category(self):
if request.method == 'GET':
return render_template('add_category.html', session=self.session)
elif request.method == 'POST':
category = {
'name': request.form['name']
}
add_category(category)
return redirect(url_for('categories'))
else:
return redirect(url_for('categories'))
@login_required
def my_projects(self):
project_list = get_projects_by_email(self.session['claims']['email'])
return render_template('my_projects.html', session=self.session, project_list=project_list)
@login_required
def add_project(self):
if request.method == 'GET':
job_list = get_jobs(page=1, per_page=50, category_id=None,
user_email=self.session['claims']['email'], freelancer_flag=False).items
return render_template('add_project.html', session=self.session, job_list=job_list)
elif request.method == 'POST':
project = {
'deadline': datetime.datetime.strptime(request.form['deadline'], '%m/%d/%Y').isoformat(),
'freelancer_email': self.session['claims']['email'],
'job_id': request.form['job_id'],
'created_at': datetime.datetime.utcnow().isoformat()
}
add_project(project)
return redirect(url_for('my_projects'))
else:
return redirect(url_for('my_projects'))
def see_project(self, project_id):
project = get_project_by_id(project_id)
return render_template('project.html', session=self.session, project=project)
@staticmethod
def delete_project(project_id):
delete_project(project_id)
return redirect(url_for('my_projects'))
@login_required
def my_jobs(self):
job_list = get_jobs(page=1, per_page=50, category_id=None,
user_email=self.session['claims']['email'], freelancer_flag=False).items
return render_template('my_jobs.html', session=self.session, job_list=job_list)
def see_job(self, job_id):
job = get_job(job_id)
return render_template('job.html', session=self.session, job=job)
@staticmethod
def delete_job(job_id):
delete_job(job_id)
return redirect(url_for('my_jobs'))
@login_required
def add_job(self):
if request.method == 'GET':
category_list = get_categories()
return render_template('add_job.html', session=self.session, category_list=category_list)
elif request.method == 'POST':
job = {
'user_email': self.session['claims']['email'],
'title': request.form['title'],
'description': request.form['description'],
'payment': request.form['payment'],
'created_at': datetime.datetime.utcnow().isoformat(),
'category_id': request.form['category_id'],
}
add_job(job)
return redirect(url_for('my_jobs'))
else:
return redirect(url_for('my_jobs'))
def marketplace(self):
categories_list = get_categories()
page = request.args.get('page', 1, type=int)
per_page = request.args.get('per_page', 10, type=int)
category_id = request.args.get('category_id', categories_list[0].id, type=int)
category_name = request.args.get('category_name', categories_list[0].name, type=str)
product_list = get_marketplace_projects(page, per_page, category_id, self.session['claims']['email'], False)
return render_template('marketplace.html', session=self.session, products=product_list,
categories=categories_list, current_category_id=category_id,
current_category_name=category_name)
def product(self, product_id):
product = get_marketplace_project(product_id)
return render_template('product_page.html', session=self.session, product=product)
@staticmethod
def unauthorized_handler():
return redirect(url_for('login'))
@staticmethod
def not_found(e):
return str(e)
@staticmethod
def server_error(e):
logging.exception('An error occurred during a request.')
return """
An internal error occurred: <pre>{}</pre>
See logs for full stacktrace.
""".format(e), 500
@sitemap_page_needed.connect
def create_page(self, page, urlset):
self.cache[page] = self.flask_sitemap.render_page(urlset=urlset)
def load_page(self, fn):
@wraps(fn)
def loader(*args, **kwargs):
page = kwargs.get('page')
data = self.cache.get(page)
return data if data else fn(*args, **kwargs)
return loader
def test_default_config(self):
Sitemap(app=self.app)
for k in dir(default_config):
if k.startswith('SITEMAP_'):
assert self.app.config.get(k) == getattr(default_config, k)
from flask_login import LoginManager
from flask_sqlalchemy import SQLAlchemy
from flask_sitemap import Sitemap
from flask_login import LoginManager
from flask_migrate import Migrate
class AppHelper(object):
def init_app(self, app, db=None, directory=None, **kwargs):
self.app = app
db = SQLAlchemy()
sitemap = Sitemap()
login_manager = LoginManager()
migrate = Migrate()
app_helper = AppHelper()
@login_manager.user_loader
def load_user(user_id):
from app.models import User
return User.query.get(int(user_id))
login_manager.session_protection = 'strong'
login_manager.login_view = 'admin.login'
# login_manager.login_message = 'Your custom message'
login_manager.login_message_category = 'warning'
# Shop Classes/Libraries
#
from Shop.Wallpaper import Wallpaper
#
# Firebase Python Classes
#
from wtforms import Form, BooleanField, StringField, PasswordField, validators
import Login.Login as FB
from flask_socketio import SocketIO
#
# GitIgnore'd
#
import Login.DBDetails as Settings
app = Flask(__name__)
smp = Sitemap(app=app)
PICKLE_FILE = "keypair.hypercharged"
carEvents = []
settings = {"Home": {"description": "Hello"}}
secret, publish = "", ""
socketio = SocketIO(app)
"""
KEY FOR DEVS: DO NOT, UNDER ANY CIRCUMSTANCE, LEAK SECRET OR PUBLISHABLE KEYS --> Stored in keypair.hypercharged file
"""
try:
pick = pickle.load(open(PICKLE_FILE, 'rb'))
publish = pick["PUBLISHABLE_KEY"]
secret = pick["SECRET_KEY"]
except FileNotFoundError:
from flask import Flask, render_template, jsonify, send_from_directory, session, redirect, request, make_response, g, url_for
from flask_sitemap import Sitemap
from logger import set_debug
app = Flask(__name__)
# have to update the Jinja template strings so they don't conflict w/Vue
jinja_options = app.jinja_options.copy() # immutable dict
jinja_options['variable_start_string'] = '(('
jinja_options['variable_end_string'] = '))'
app.jinja_options = jinja_options
sitemap = Sitemap(app=app)
@app.before_request
def before_request():
pass
@app.teardown_request
def teardown_request(exception):
pass
def render(template, title, vue=(), **kwargs):
"""
Helper function to render templates given a page title
from flask import Flask from flask_sitemap import Sitemap app = Flask(__name__) ext = Sitemap(app=app) from Kolektif._endpointler import anaSayfa, _hata, eczane, haber, udemy, deprem, akaryakit, hava, bim, doviz
from flask_login import LoginManager, AnonymousUserMixin
from flask_mail import Mail
from flask_dropzone import Dropzone
from flask_wtf import CSRFProtect
from flask_sitemap import Sitemap
from flask_avatars import Avatars
from flask_migrate import Migrate
from flask_moment import Moment
db = SQLAlchemy()
bootstrap = Bootstrap()
login_manager = LoginManager()
mail = Mail()
dropzone = Dropzone()
csrf = CSRFProtect()
ext = Sitemap()
avatars = Avatars()
migrate = Migrate()
moment = Moment()
@login_manager.user_loader
def load_user(user_id):
from app.models import User
user = User.query.get(int(user_id))
return user
class Guest(AnonymousUserMixin):
@property
def is_admin(self):
