def update_card_position(self, data):
data = json.loads(data)
cols = {}
for col in self.columns:
cols[col().id] = (col(), col)
orig, __ = cols[data['orig']]
dest, dest_comp = cols[data['dest']]
card_comp = orig.remove_card_by_id(data['card'])
dest.insert_card_comp(dest_comp, data['index'], card_comp)
card = card_comp()
values = {
'from': orig.get_title(),
'to': dest.get_title(),
'card': card.get_title()
}
self.action_log.for_card(card).add_history(security.get_user(),
u'card_move', values)
# reindex it in case it has been moved to the archive column
scard = fts_schema.Card(**card.to_document())
self.search_engine.update_document(scard)
self.search_engine.commit()
session.flush()
def create_template_empty():
board = DataBoard(title=u'Empty board', is_template=True, visibility=1)
board.weight_config = DataBoardWeightConfig()
for title, color in DEFAULT_LABELS:
board.create_label(title=title, color=color)
session.flush()
return board
def move_cards(self, v):
"""Function called after drag and drop of a card or column
In:
- ``v`` -- a structure containing the lists/cards position
(ex . [ ["list_1", ["card_1", "card_2"]],
["list_2", ["card_3", "card_4"]] ])
"""
security.check_permissions('edit', self)
ids = json.loads(v)
cards = {}
cols = {}
for col in self.columns:
cards.update(dict([(card().id, card) for card in col().cards
if not isinstance(card(), popin.Empty)]))
cols[col().id] = col
# move columns
self.columns = []
for (col_index, (col_id, card_ids)) in enumerate(ids):
comp_col = cols[col_id]
self.columns.append(comp_col)
comp_col().change_index(col_index)
comp_col().move_cards([cards[id_] for id_ in card_ids])
session.flush()
def insert_card(self, index, card):
done = False
if card not in self.cards:
self.cards.insert(index, card)
session.flush()
done = True
return done
def update_user(self):
# checks the other fields and updates data
if super(UserLineEditor, self).commit((), ('corporation_id', 'direction_id', 'site_id', 'service_id', 'fi_uid', 'enabled')):
# FIXME: use the OrganizationRepository
get_orga = lambda id: OrganizationData.get(
id) if id != -1 else None
u = self.user_repository.get_by_uid(self.uid)
u.corporation = get_orga(self.corporation_id())
u.direction = get_orga(self.direction_id())
u.service = get_orga(self.service_id())
u.site = get_orga(self.site_id())
u.subsite = get_orga(self.subsite_id())
u.enabled = self.enabled()
u.fi_uid = self.fi_uid()
session.flush()
self.orig_corp_id = u.corporation_id or -1L
self.orig_dir_id = u.direction_id or -1L
self.orig_service_id = u.service_id or -1L
self.orig_site_id = u.site_id or -1L
self.orig_subsite_id = u.subsite_id or -1L
self.orig_enabled = u.enabled
self.orig_fi_uid = u.fi_uid
flashmessage.set_flash(_(u'User modified'))
return True
def copy(self, parent):
new_data = DataColumn(title=self.title,
index=self.index,
nb_max_cards=self.nb_max_cards,
board=parent)
session.flush()
return new_data
def copy(self, parent):
new_data = DataLabel(title=self.title,
color=self.color,
index=self.index,
board=parent)
session.flush()
return new_data
def move_cards(self, v):
"""Function called after drag and drop of a card or column
In:
- ``v`` -- a structure containing the lists/cards position
(ex . [ ["list_1", ["card_1", "card_2"]],
["list_2", ["card_3", "card_4"]] ])
"""
security.check_permissions('edit', self)
ids = json.loads(v)
cards = {}
cols = {}
for col in self.columns:
cards.update(dict([(card().id, card) for card in col().cards
if not isinstance(card(), popin.Empty)]))
cols[col().id] = col
# move columns
self.columns = []
for (col_index, (col_id, card_ids)) in enumerate(ids):
comp_col = cols[col_id]
self.columns.append(comp_col)
comp_col().change_index(col_index)
comp_col().move_cards([cards[id_] for id_ in card_ids])
session.flush()
def populate_states():
workflow = get_workflow()
WFSteps = workflow.WFSteps
WFStates = workflow.WFStates
states = (
(WFStates.INITIAL_STATE, WFSteps.NO_STEP),
(WFStates.DRAFT_STATE, WFSteps.NO_STEP),
(WFStates.FI_NORMALIZE_STATE, WFSteps.SUBMITTED_STEP),
(WFStates.AUTHOR_NORMALIZE_STATE, WFSteps.SUBMITTED_STEP),
(WFStates.DI_APPRAISAL_STATE, WFSteps.STUDY_STEP),
(WFStates.RETURNED_BY_DI_STATE, WFSteps.STUDY_STEP),
(WFStates.DI_APPROVED_STATE, WFSteps.SUGGESTED_STEP),
(WFStates.SELECTED_STATE, WFSteps.SELECTED_STEP),
(WFStates.PROJECT_STATE, WFSteps.PROJECT_STEP),
(WFStates.PROTOTYPE_STATE, WFSteps.PROTOTYPE_STEP),
(WFStates.EXTENDED_STATE, WFSteps.EXTENDED_STEP),
(WFStates.FI_REFUSED_STATE, WFSteps.SUBMITTED_STEP),
(WFStates.PROTOTYPE_REFUSED_STATE, WFSteps.PROTOTYPE_STEP),
(WFStates.DI_REFUSED_STATE, WFSteps.STUDY_STEP),
(WFStates.PROJECT_REFUSED_STATE, WFSteps.PROJECT_STEP),
(WFStates.SELECT_REFUSED_STATE, WFSteps.SELECTED_STEP),
(WFStates.APPROVAL_REFUSED_STATE, WFSteps.SUGGESTED_STEP),
(WFStates.DSIG_BASKET_STATE, WFSteps.NO_STEP),
)
for label, step in states:
step = StepData.get_by(label=step)
StateData(label=label, step=step)
session.flush()
def update_card_position(self, data):
security.check_permissions('edit', self)
data = json.loads(data)
cols = {}
for col in self.columns:
cols[col().id] = col()
orig = cols[data['orig']]
if data['orig'] != data['dest']: # Move from one column to another
dest = cols[data['dest']]
card = orig.remove_card(data['card'])
dest.insert_card(card, data['index'])
values = {'from': orig.title().text,
'to': dest.title().text,
'card': card().data.title}
notifications.add_history(self.data, card().data,
security.get_user().data,
u'card_move', values)
# reindex it in case it has been moved to the archive column
scard = fts_schema.Card.from_model(card().data)
self.search_engine.update_document(scard)
self.search_engine.commit()
else: # Reorder only
orig.move_card(data['card'], data['index'])
session.flush()
def test_get_by_role(self):
user = create_user()
create_user()
user.add_role(RoleType.Facilitator)
session.flush()
facilitators = self.user_repository.get_facilitators()
self.assertTrue(user in facilitators)
def update_card_position(self, data):
security.check_permissions('edit', self)
data = json.loads(data)
cols = {}
for col in self.columns:
cols[col().id] = col()
orig = cols[data['orig']]
if data['orig'] != data['dest']: # Move from one column to another
dest = cols[data['dest']]
card = orig.remove_card(data['card'])
dest.insert_card(card, data['index'])
values = {'from': orig.title().text,
'to': dest.title().text,
'card': card().data.title}
notifications.add_history(self.data, card().data,
security.get_user().data,
u'card_move', values)
# reindex it in case it has been moved to the archive column
scard = fts_schema.Card.from_model(card().data)
self.search_engine.update_document(scard)
self.search_engine.commit()
else: # Reorder only
orig.move_card(data['card'], data['index'])
session.flush()
def insert_card(self, index, card):
done = False
if card not in self.cards:
self.cards.insert(index, card)
session.flush()
done = True
return done
def add(self, v):
"""Add a new comment to the card
In:
- ``v`` -- the comment content
"""
security.check_permissions("comment", self.parent)
if v is None:
return
v = v.strip()
if v:
v = validator.clean_text(v)
user = security.get_user()
comment = DataComment(
comment=v.strip(),
card_id=self.parent.data.id,
author=user.data,
creation_date=datetime.datetime.utcnow(),
)
session.add(comment)
session.flush()
data = {"comment": v.strip(), "card": self.parent.title().text}
notifications.add_history(
self.parent.column.board.data, self.parent.data, security.get_user().data, u"card_add_comment", data
)
self.comments.insert(0, self._create_comment_component(comment))
def add(self, v):
"""Add a new comment to the card
In:
- ``v`` -- the comment content
"""
security.check_permissions('comment', self.parent)
if v is None:
return
v = v.strip()
if v:
v = validator.clean_text(v)
user = security.get_user()
comment = DataComment(comment=v.strip(),
card_id=self.parent.data.id,
author=user.data,
creation_date=datetime.datetime.utcnow())
session.add(comment)
session.flush()
data = {'comment': v.strip(), 'card': self.parent.title().text}
notifications.add_history(self.parent.column.board.data,
self.parent.data,
security.get_user().data,
u'card_add_comment', data)
self.comments.insert(0, self._create_comment_component(comment))
def test_get_by_role(self):
user = create_user()
create_user()
user.add_role(RoleType.Facilitator)
session.flush()
facilitators = self.user_repository.get_facilitators()
self.assertTrue(user in facilitators)
def update_user(self):
# checks the other fields and updates data
if super(UserLineEditor, self).commit(
(), ('corporation_id', 'direction_id', 'site_id', 'service_id',
'fi_uid', 'enabled')):
# FIXME: use the OrganizationRepository
get_orga = lambda id: OrganizationData.get(id
) if id != -1 else None
u = self.user_repository.get_by_uid(self.uid)
u.corporation = get_orga(self.corporation_id())
u.direction = get_orga(self.direction_id())
u.service = get_orga(self.service_id())
u.site = get_orga(self.site_id())
u.subsite = get_orga(self.subsite_id())
u.enabled = self.enabled()
u.fi_uid = self.fi_uid()
session.flush()
self.orig_corp_id = u.corporation_id or -1L
self.orig_dir_id = u.direction_id or -1L
self.orig_service_id = u.service_id or -1L
self.orig_site_id = u.site_id or -1L
self.orig_subsite_id = u.subsite_id or -1L
self.orig_enabled = u.enabled
self.orig_fi_uid = u.fi_uid
flashmessage.set_flash(_(u'User modified'))
return True
def create_template_empty():
board = DataBoard(title=u'Empty board', is_template=True, visibility=1)
board.weight_config = DataBoardWeightConfig()
for title, color in DEFAULT_LABELS:
board.create_label(title=title, color=color)
session.flush()
return board
def populate_users(default_users):
"""Populate the default users (special accounts)"""
facilitator_uid = default_users['FACILITATOR']
developer_uid = default_users['DEVELOPER']
admin_uid = default_users['ADMIN']
innovator_uid = default_users['INNOVATOR']
special_users = (
(facilitator_uid, facilitator_uid, facilitator_uid.capitalize(),
facilitator_uid.capitalize(),
u'{}@eureka-open.com'.format(facilitator_uid), facilitator_uid,
[RoleType.Facilitator]),
(developer_uid, developer_uid, u'Ideas', developer_uid.capitalize(),
u'{}@eureka-open.com'.format(developer_uid), facilitator_uid,
[RoleType.Developer]),
(admin_uid, admin_uid, admin_uid.capitalize(), u'Eureka',
u'{}@eureka-open.com'.format(admin_uid), facilitator_uid,
[RoleType.DSIG]),
(innovator_uid, innovator_uid, innovator_uid.capitalize(), u'User',
u'{}@eureka-open.com'.format(innovator_uid), facilitator_uid, []),
)
for (uid, password, firstname, lastname, email, fi_uid,
roles) in special_users:
user = UserRepository().create(uid=uid,
password=password,
firstname=firstname,
lastname=lastname,
email=email,
fi_uid=fi_uid)
for role in roles:
user.add_role(role)
session.flush()
def populate_states():
workflow = get_workflow()
WFSteps = workflow.WFSteps
WFStates = workflow.WFStates
states = (
(WFStates.INITIAL_STATE, WFSteps.NO_STEP),
(WFStates.DRAFT_STATE, WFSteps.NO_STEP),
(WFStates.FI_NORMALIZE_STATE, WFSteps.SUBMITTED_STEP),
(WFStates.AUTHOR_NORMALIZE_STATE, WFSteps.SUBMITTED_STEP),
(WFStates.DI_APPRAISAL_STATE, WFSteps.STUDY_STEP),
(WFStates.RETURNED_BY_DI_STATE, WFSteps.STUDY_STEP),
(WFStates.DI_APPROVED_STATE, WFSteps.SUGGESTED_STEP),
(WFStates.SELECTED_STATE, WFSteps.SELECTED_STEP),
(WFStates.PROJECT_STATE, WFSteps.PROJECT_STEP),
(WFStates.PROTOTYPE_STATE, WFSteps.PROTOTYPE_STEP),
(WFStates.EXTENDED_STATE, WFSteps.EXTENDED_STEP),
(WFStates.FI_REFUSED_STATE, WFSteps.SUBMITTED_STEP),
(WFStates.PROTOTYPE_REFUSED_STATE, WFSteps.PROTOTYPE_STEP),
(WFStates.DI_REFUSED_STATE, WFSteps.STUDY_STEP),
(WFStates.PROJECT_REFUSED_STATE, WFSteps.PROJECT_STEP),
(WFStates.SELECT_REFUSED_STATE, WFSteps.SELECTED_STEP),
(WFStates.APPROVAL_REFUSED_STATE, WFSteps.SUGGESTED_STEP),
(WFStates.DSIG_BASKET_STATE, WFSteps.NO_STEP),
)
for label, step in states:
step = StepData.get_by(label=step)
StateData(label=label, step=step)
session.flush()
def copy(self, parent):
new_data = DataLabel(title=self.title,
color=self.color,
index=self.index,
board=parent)
session.flush()
return new_data
def create_template_todo():
board = DataBoard(title=u'Todo', is_template=True, visibility=1)
for index, title in enumerate((u'To Do', u'Doing', u'Done')):
board.create_column(index, title)
for title, color in DEFAULT_LABELS:
board.create_label(title=title, color=color)
session.flush()
return board
def create_template_todo():
board = DataBoard(title=u'Todo', is_template=True, visibility=1)
for index, title in enumerate((u'To Do', u'Doing', u'Done')):
board.create_column(index, title)
for title, color in DEFAULT_LABELS:
board.create_label(title=title, color=color)
session.flush()
return board
def append_card(self, card):
done = False
if card not in self.cards:
card.index = None
self.cards.append(card)
session.flush()
done = True
return done
def test_photo(self):
user = create_user()
photo = self.create_photo(50)
user.photo = photo
session.flush()
self.assertTrue(user.photo is not None)
user.photo = None
self.assertTrue(user.photo is None)
def test_photo(self):
user = create_user()
photo = self.create_photo(50)
user.photo = photo
session.flush()
self.assertTrue(user.photo is not None)
user.photo = None
self.assertTrue(user.photo is None)
def append_card(self, card):
done = False
if card not in self.cards:
card.index = None
self.cards.append(card)
session.flush()
done = True
return done
def test_remove_existing_role(self):
user = create_user()
user.add_role(RoleType.Facilitator)
session.flush()
user.remove_role(RoleType.Facilitator)
self.assert_(not user.has_role(RoleType.Facilitator))
session.flush()
facilitators = self.user_repository.get_facilitators()
self.assert_(user not in facilitators)
def test_change_password_success(self):
user = create_user(uid=u'jdoe',
firstname=u'John',
lastname=u'Doe',
password=u'password')
new_password = u'newpass'
user.change_password(new_password)
session.flush()
self.assertEqual(user.password, user.encrypt_password(new_password))
def test_remove_existing_role(self):
user = create_user()
user.add_role(RoleType.Facilitator)
session.flush()
user.remove_role(RoleType.Facilitator)
self.assert_(not user.has_role(RoleType.Facilitator))
session.flush()
facilitators = self.user_repository.get_facilitators()
self.assert_(user not in facilitators)
def create_board():
"""Create boards with default columns and default cards
"""
user_test = get_or_create_data_user()
data_board = boardsmanager.BoardsManager().create_board(word(), user_test,
True)
session.add(data_board)
session.flush()
assets_manager = DummyAssetsManager()
return board.Board(data_board.id, 'boards', '', '', '', assets_manager, None)
def test_enabled(self):
uid = u'jdoe'
create_user(uid=uid)
session.flush()
user = self.user_repository.get_by_uid(uid)
self.assert_(user.enabled) # this is what we want most of the time
user.enabled = False
session.flush()
user = self.user_repository.get_by_uid(uid)
self.assert_(not user.enabled)
def import_article_topics_csv(csv_file):
reader = UnicodeDictReader(csv_file, delimiter=';')
for row in reader:
article = ArticleTopicData()
article.label = row['label'].strip()
article.key = row['key'].strip()
article.default = int(row['default'].strip())
session.flush()
def import_article_topics_csv(csv_file):
reader = UnicodeDictReader(csv_file, delimiter=';')
for row in reader:
article = ArticleTopicData()
article.label = row['label'].strip()
article.key = row['key'].strip()
article.default = int(row['default'].strip())
session.flush()
def delete_comment(self, comp):
"""Delete a comment.
In:
- ``comment`` -- the comment to delete
"""
self.comments.remove(comp)
comment = comp()
DataComment.get(comment.db_id).delete()
session.flush()
def create_board():
"""Create boards with default columns and default cards
"""
user_test = get_or_create_data_user()
data_board = boardsmanager.BoardsManager().create_board(word(), user_test,
True)
session.add(data_board)
session.flush()
_services = create_services()
return _services(board.Board, data_board.id, 'boards', '', '', None)
def test_enabled(self):
uid = u'jdoe'
create_user(uid=uid)
session.flush()
user = self.user_repository.get_by_uid(uid)
self.assert_(user.enabled) # this is what we want most of the time
user.enabled = False
session.flush()
user = self.user_repository.get_by_uid(uid)
self.assert_(not user.enabled)
def test_set_firstname_and_lastname(self):
uid = u'jdoe'
user = create_user(uid=uid)
firstname = u'John'
lastname = u'Doe'
user.firstname = firstname
user.lastname = lastname
session.flush()
user = self.user_repository.get_by_uid(uid)
self.assertEquals(user.firstname, firstname)
self.assertEquals(user.lastname, lastname)
def import_domains_csv(csv_file):
reader = UnicodeDictReader(csv_file, delimiter=';')
for line in reader:
d = DomainData()
d.label = line['label'].strip()
d.rank = int(line['rank'].strip())
d.en_label = line['en_label'].strip()
d.fr_label = line['fr_label'].strip()
session.flush()
def test_change_password_success(self):
user = create_user(
uid=u'jdoe',
firstname=u'John',
lastname=u'Doe',
password=u'password'
)
new_password = u'newpass'
user.change_password(new_password)
session.flush()
self.assertEqual(user.password, user.encrypt_password(new_password))
def test_deleting_a_user_also_remove_her_roles(self):
user = create_user()
user.add_role(RoleType.Facilitator)
session.flush()
user_uid = user.uid
user.delete()
session.flush()
facilitators_uids = [
u.uid for u in self.user_repository.get_facilitators()
]
self.assertFalse(user_uid in facilitators_uids)
def test_creation(self):
uid = u'jdoe'
firstname = u'John'
lastname = u'Doe'
create_user(uid=uid, firstname=firstname, lastname=lastname)
session.flush()
user = self.user_repository.get_by_uid(uid)
self.assertEquals(user.uid, uid)
self.assertEquals(user.firstname, firstname)
self.assertEquals(user.lastname, lastname)
self.assert_(user.enabled)
def remove_invitation(self, email):
""" Remove invitation
In:
- ``email`` -- guest email to invalidate
"""
for token in self.data.pending:
if token.username == email:
token.delete()
session.flush()
break
def copy(self, parent):
new_data = DataBoard(title=self.title,
description=self.description,
background_position=self.background_position,
title_color=self.title_color,
comments_allowed=self.comments_allowed,
votes_allowed=self.votes_allowed,
weighting_cards=self.weighting_cards,
weights=self.weights)
session.flush()
return new_data
def test_deleting_a_user_also_remove_her_roles(self):
user = create_user()
user.add_role(RoleType.Facilitator)
session.flush()
user_uid = user.uid
user.delete()
session.flush()
facilitators_uids = [
u.uid for u in self.user_repository.get_facilitators()
]
self.assertFalse(user_uid in facilitators_uids)
def copy(self, parent):
new_data = DataBoard(title=self.title,
description=self.description,
background_position=self.background_position,
title_color=self.title_color,
comments_allowed=self.comments_allowed,
votes_allowed=self.votes_allowed,
weighting_cards=self.weighting_cards,
weights=self.weights)
session.flush()
return new_data
def remove_invitation(self, email):
""" Remove invitation
In:
- ``email`` -- guest email to invalidate
"""
for token in self.data.pending:
if token.username == email:
token.delete()
session.flush()
break
def test_creation(self):
uid = u'jdoe'
firstname = u'John'
lastname = u'Doe'
create_user(uid=uid, firstname=firstname, lastname=lastname)
session.flush()
user = self.user_repository.get_by_uid(uid)
self.assertEquals(user.uid, uid)
self.assertEquals(user.firstname, firstname)
self.assertEquals(user.lastname, lastname)
self.assert_(user.enabled)
def test_set_firstname_and_lastname(self):
uid = u'jdoe'
user = create_user(uid=uid)
firstname = u'John'
lastname = u'Doe'
user.firstname = firstname
user.lastname = lastname
session.flush()
user = self.user_repository.get_by_uid(uid)
self.assertEquals(user.firstname, firstname)
self.assertEquals(user.lastname, lastname)
def import_domains_csv(csv_file):
reader = UnicodeDictReader(csv_file, delimiter=';')
for line in reader:
d = DomainData()
d.label = line['label'].strip()
d.rank = int(line['rank'].strip())
d.en_label = line['en_label'].strip()
d.fr_label = line['fr_label'].strip()
session.flush()
def populate_steps():
wfsteps = get_workflow().WFSteps
steps = ((wfsteps.NO_STEP, 100), (wfsteps.SUBMITTED_STEP, 1),
(wfsteps.STUDY_STEP, 2), (wfsteps.SUGGESTED_STEP, 3),
(wfsteps.SELECTED_STEP, 4), (wfsteps.PROJECT_STEP, 5),
(wfsteps.PROTOTYPE_STEP, 6), (wfsteps.EXTENDED_STEP, 7))
for label, rank in steps:
StepData(label=label, rank=rank)
session.flush()
def import_improvement_domains_csv(csv_file):
reader = UnicodeDictReader(csv_file, delimiter=';')
for line in reader:
d = ImprovementDomainData.query.filter(
ImprovementDomainData.label == line['label']).first()
if d is None:
d = ImprovementDomainData()
d.label = line['label'].strip()
d.rank = int(line['rank'].strip())
session.flush()
def import_improvement_domains_csv(csv_file):
reader = UnicodeDictReader(csv_file, delimiter=';')
for line in reader:
d = ImprovementDomainData.query.filter(
ImprovementDomainData.label == line['label']).first()
if d is None:
d = ImprovementDomainData()
d.label = line['label'].strip()
d.rank = int(line['rank'].strip())
session.flush()
def add_member(self, new_member, role='member'):
""" Add new member to the board
In:
- ``new_member`` -- user to add (DataUser instance)
- ``role`` -- role's member (manager or member)
"""
self.board_members.append(DataBoardMember(member=new_member.data))
if role == 'manager':
self.managers.append(new_member.data)
session.flush()
def update_column_position(self, data):
data = json.loads(data)
cols = []
found = None
for col in self.columns:
if col().id == data['list']:
found = col
else:
cols.append(col)
cols.insert(data['index'], found)
for i, col in enumerate(cols):
col().change_index(i)
self.columns = cols
session.flush()
def update_column_position(self, data):
data = json.loads(data)
cols = []
found = None
for col in self.columns:
if col().id == data['list']:
found = col
else:
cols.append(col)
cols.insert(data['index'], found)
for i, col in enumerate(cols):
col().change_index(i)
self.columns = cols
session.flush()
def update_column_position(self, data):
security.check_permissions("edit", self)
data = json.loads(data)
cols = []
found = None
for col in self.columns:
if col().id == data["list"]:
found = col
else:
cols.append(col)
cols.insert(data["index"], found)
for i, col in enumerate(cols):
col().change_index(i)
self.columns = cols
session.flush()
