Imgee V2

.gitignore

@@ -20,3 +20,7 @@ baseframe-packed.css
 baseframe-packed.js
 error.log
 imgee/static/uploads
+*.bak
+instance/production.env.sh
+imgee/static/gen
+tests/imgee/

.travis.yml

@@ -1,5 +1,5 @@
-env: 
-  global: 
+env:
+  global:
   - secure: |-
       G5Dn+zkbN/BeNoopxtM2idC2Hy1ebJxRxprD7XEAmH6VO26ANgWYLI1dMrdH
       uQc9F317STawQ/Um6KnqjErOKkC2BUYTOTj8AzoPVFz6NcK/Ca4d9Vfbtf5u
@@ -8,17 +8,23 @@ env:
       SQ9lTuSD5jg3NSM8yMfMU58ppAYBgd+6VQP01wUPFrV/JSsbfgnVjMTm/Nbk
       EGeHYvQUiyAg7zM4KdNJr6txj+jBE8MAeh7EwYNHoh9B7Vx//GxmXFnWyjXV
       9cJkFroDW1Zfs2SZjLtzMQC8YXE30jmMxg+XHCQewKzRa4u1320=
+  # this is already being set in runtests.sh,
+  # but need to set it for `./manage.py init` too.
+  # could set it before running manage.py, but chose to put it here
+  - FLASK_ENV=TESTING
 language: python
-python: 
+python:
   - 2.7
-before_script:
-  - mkdir imgee/static/test_uploads
-script: 
-  - nosetests --with-coverage
-install: 
-  - sudo apt-get -qq install zlib1g-dev libfreetype6-dev liblcms1-dev libwebp-dev libjpeg-dev libpng-dev libfreetype6-dev libtiff4-dev librsvg2-dev ghostscript imagemagick pandoc
+services:
+  - redis-server
+install:
+  - sudo apt-get -qq install zlib1g-dev libfreetype6-dev liblcms1-dev libwebp-dev libjpeg-dev libpng-dev libfreetype6-dev libtiff4-dev librsvg2-dev ghostscript imagemagick pandoc inkscape
   - pip install -r requirements.txt
-  - pip install nose coverage BeautifulSoup4 Pillow
+  - pip install -r test_requirements.txt
+before_script:
+  - ./manage.py init  # creates the test upload directory
+script:
+  - ./runtests.sh
 notifications:
   email: false
   slack:

imgee/__init__.py

@@ -3,7 +3,6 @@
 # The imports in this file are order-sensitive
 
 import os
-from celery import Celery
 
 from flask import Flask, redirect, url_for
 from flask_lastuser import Lastuser
@@ -16,41 +15,29 @@
 version = Version(__version__)
 app = Flask(__name__, instance_relative_config=True)
 lastuser = Lastuser()
-celery = Celery()
 
 assets['imgee.css'][version] = 'css/app.css'
 
 from . import models, views
 from .models import db
-from .api import api
-from .async import TaskRegistry
+from .tasks import TaskRegistry
 
 registry = TaskRegistry()
 
-def mkdir_p(dirname):
-    if not os.path.exists(dirname):
-        os.makedirs(dirname)
-
-
-# Configure the app
-
+# Configure the application
 coaster.app.init_app(app)
 migrate = Migrate(app, db)
 baseframe.init_app(app, requires=['baseframe', 'picturefill', 'imgee'])
 lastuser.init_app(app)
 lastuser.init_usermanager(UserManager(db, models.User))
+registry.init_app(app)
 
 
 @app.errorhandler(403)
 def error403(error):
     return redirect(url_for('login'))
 
-if app.config.get('MEDIA_DOMAIN') and (
-        app.config['MEDIA_DOMAIN'].startswith('http:') or
-        app.config['MEDIA_DOMAIN'].startswith('https:')):
+if app.config.get('MEDIA_DOMAIN', '').lower().startswith(('http://', 'https://')):
     app.config['MEDIA_DOMAIN'] = app.config['MEDIA_DOMAIN'].split(':', 1)[1]
-mkdir_p(app.config['UPLOADED_FILES_DEST'])
 
-celery.conf.add_defaults(app.config)
-registry.set_connection()
-app.register_blueprint(api, url_prefix='/api/1')
+app.upload_folder = os.path.join(app.static_folder, app.config.get('UPLOADED_FILES_DIR'))

imgee/forms.py

@@ -1,19 +1,17 @@
 # -*- coding: utf-8 -*-
 
-import os.path
 from coaster.utils import make_name
-from flask_wtf import Form
+from baseframe.forms import Form
 from wtforms.validators import Required, ValidationError, Length
-from wtforms import (FileField, TextField, HiddenField,
-        SelectMultipleField, SelectField)
+from wtforms import (FileField, TextField, HiddenField, SelectField)
 
 from imgee import app
 from imgee.models import Label
-from imgee.utils import get_file_type, is_file_allowed
+from imgee.utils import is_file_allowed
 
 
 def valid_file(form, field):
-    if not is_file_allowed(field.data.stream):
+    if not is_file_allowed(field.data.stream, field.data.mimetype, field.data.filename):
         raise ValidationError("Sorry, unknown image format. Please try uploading another file.")
 
 
@@ -28,8 +26,9 @@ class DeleteImageForm(Form):
 class PurgeCacheForm(Form):
     pass
 
+
 def reserved_words():
-    """get all words which can't be used as labels"""
+    """Get all words which can't be used as labels"""
     words = []
     for rule in app.url_map.iter_rules():
         r = rule.rule
@@ -67,6 +66,7 @@ class EditTitleForm(Form):
     file_name = HiddenField('file_name')
     file_title = TextField('title', validators=[Required(), Length(max=250)])
 
+
 class UpdateTitle(Form):
     title = TextField('Title', validators=[Required(), Length(max=250)])
 

imgee/models/stored_file.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 
 from coaster.sqlalchemy import BaseNameMixin, BaseScopedNameMixin
-import imgee
+from imgee import app, url_for
 from imgee.models import db
 from imgee.utils import newid, guess_extension
 
@@ -58,7 +58,33 @@ def __repr__(self):
     def extn(self):
         return guess_extension(self.mimetype, self.orig_extn) or ''
 
-    def is_queued_for_deletion(self):
-        if imgee.app.config.get('CELERY_ALWAYS_EAGER'):
-            return False
-        return imgee.registry.is_queued_for_deletion(self.name+self.extn)
+    @property
+    def filename(self):
+        return self.name + self.extn
+
+    def dict_data(self):
+        return dict(
+            title=self.title,
+            uploaded=self.created_at.isoformat() + 'Z',
+            filesize=app.jinja_env.filters['filesizeformat'](self.size),
+            imgsize=u'%s×%s px' % (self.width, self.height),
+            url=url_for('view_image', profile=self.profile.name, image=self.name),
+            thumb_url=url_for('get_image', image=self.name, size=app.config.get('THUMBNAIL_SIZE'))
+        )
+
+    def add_labels(self, labels):
+        new_labels = set(labels)
+        old_labels = set(self.labels)
+        if new_labels != old_labels:
+            self.labels = labels
+
+        if new_labels == old_labels:
+            status, diff = '0', []
+        elif new_labels > old_labels:
+            status, diff = '+', (new_labels - old_labels)
+        elif (old_labels > new_labels):
+            status, diff = '-', (old_labels - new_labels)
+        else:
+            status, diff = '', new_labels
+
+        return status, list(diff)