/
tasks.py
69 lines (59 loc) · 1.82 KB
/
tasks.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import os
from skimage.io import imread
from invoke import task
from app import db, Image, Match, User, Classification
from img_serve import get_all_imgs
from sqlalchemy import distinct
@task
def create_db(ctx):
#Classification.__table__.drop(db.engine)
db.create_all()
@task
def create_users(ctx):
#User.new(name='admin', pwd='admin')
#User.new(name='kegl', pwd='kegl')
#User.new(name='mehdi', pwd='mehdi')
#User.new(name='anonymous', pwd='anonymous')
#User.new(name='akin', pwd='akin')
User.new(name='balazs', pwd='balazs')
@task
def insert_urls(ctx, filename):
for i, url in enumerate(open(filename).readlines()):
url = url[0:-1]
db.session.add(Image(url=url))
db.session.commit()
@task
def insert_local(ctx, folder='imgs', pattern=''):
for filename in get_all_imgs(folder, pattern=pattern):
if not accept(filename):
print("{} not accepted".format(filename))
continue
url = '{LOCAL}/' + filename
try:
db.session.add(Image(url=url))
print("Adding {}...".format(filename))
db.session.commit()
except Exception as ex:
db.session.rollback()
print("Exception : {}, ignoring {}".format(ex, filename))
@task
def remove(ctx, pattern=''):
q = Image.query.filter(Image.url.like(pattern))
print('nb deleted : {}'.format(q.delete(synchronize_session=False)))
db.session.commit()
@task
def remove_matches(ctx, experiment=''):
q = Match.query.filter(Match.experiment==experiment)
q.delete(synchronize_session=False)
db.session.commit()
@task
def experiments(ctx):
q = db.session.query(distinct(Match.experiment))
q = list(q)
print(q)
def accept(filename):
val = imread(filename).sum()
if val > 0:
return True
else:
return False