def create_status():
if not Status.select():
status = Status(status="new")
status.save()
status = Status(status="progress")
status.save()
status = Status(status="review")
status.save()
status = Status(status="done")
status.save()
def on_post(self, req, resp):
if req.get_param('media_ids'):
user = req.context['user']
status = Status(
caption=req.get_param('status') or '',
visibility=bool(req.get_param('visibility')), #False if None
user=user,
sensitive=bool(req.get_param('sensitive')),
remote=False,
story=bool(req.get_param('is_story'))
)
if status.sensitive:
status.spoliet_text=req.get_param('spoiler_text')
status.save()
if req.get_param('media_ids') != None:
for image in req.get_param('media_ids').split(','):
m = Media.get_or_none(media_name=image)
m.status = status
m.save()
#Increment the number of posts uploaded
UserProfile.update({UserProfile.statuses_count: UserProfile.statuses_count + 1}).where(UserProfile.id == user.id).execute()
spread_status(status)
resp.status = falcon.HTTP_200
resp.body = json.dumps(status.to_json(),default=str)
elif req.get_param('in_reply_to_id'):
replying_to = Status.get_or_none(id=req.get_param('in_reply_to_id'))
if replying_to:
status = Status(
caption = req.get_param('status'),
user = user,
remote = False,
story = False,
in_reply_to = replying_to,
sensitive = replying_to.sensitive,
spoiler_text = req.get_param('spoiler_text') or replying_to.spoiler_text
)
else:
resp.status = falcon.HTTP_500
resp.body = json.dumps({"Error": "Replying to bad ID"})
else:
resp.status = falcon.HTTP_500
resp.body = json.dumps({"Error": "No photo attached"})
def on_success(self, data):
event_type = "untracked"
if data.get('direct_message'):
event_type = "Direct Message"
dm = DirectMessage(data)
self.on_direct_message(dm)
elif data.get('event') and data['event'] == 'follow':
event_type = "Follow"
follow = Follow(data)
self.on_follow(follow)
elif data.get('text'):
event_type = "Status"
status = Status(data)
# ignore RTs for now
if status.type is 'retweet':
event_type = "{0} -- Retweet".format(event_type)
else:
# if it's a reply, handle as such
# otherwise, for now, do nothing.
if status.reply_to_user():
event_type = "{0} -- Reply".format(event_type)
self.on_reply(status)
print "{0} data received. Type: {1}".format(self.__class__.__name__,
event_type)
print data
def create_status_nodes(sql_context, sc):
distinct_statuses = sql_context.sql(
"SELECT DISTINCT(Status) from projects_tbl")
distinct_statuses = distinct_statuses.rdd.filter(
lambda x: x.Status is not None)
statuses_nodes = distinct_statuses.map(lambda x: Status(x))
sc.parallelize(statuses_nodes.collect()).foreachPartition(
util.create_nodes_in)
def post(self):
if self.user:
status = self.request.get('status')
if status:
s = Status(parent = status_key(),
user = self.user,
content = status)
s.put()
self.redirect("/newsfeed")
else:
self.render("errorpage.html", error = "Status cannot be empty!")
else:
self.redirect("/login")
async def init(self, ctx):
logging.info('Start configuration for server:' + ctx.message.server.id)
discord_server = ctx.message.server
discord_user = ctx.message.author
db_user = self.session.query(User).filter(
User.discord_id == discord_user.id).first()
db_server = self.session.query(Server).filter(
Server.discord_id == discord_server.id).first()
db_status = self.session.query(Status).join(Server).filter(
Server.discord_id == ctx.message.server.id).first()
if ((db_status is not None and db_status.server_status == 2)
and not (ctx.message.author.id == db_status.user.discord_id)):
logging.info('Attemted to init the server, aborting')
await self.bot.send_message(
ctx.message.channel, 'Only ' + db_status.user.user_name +
' can initialize the bot again')
return
if (db_user is None):
new_user = User(discord_user.id, discord_user.name,
discord_user.display_name)
self.session.add(new_user)
db_user = new_user
if (db_server is None):
db_status = self.session.query(Status).join(Server).filter(
Server.discord_id == discord_server.id).first()
if (db_status is not None): self.session.delete(db_status)
new_server = Server(discord_server.id, discord_server.name, '', '',
'', '')
new_status = Status(0, db_user, new_server)
self.session.add(new_server)
self.session.add(new_status)
self.session.commit()
await self.bot.send_message(
ctx.message.channel, 'Started bot configuration for this server.')
await self.bot.send_message(
ctx.message.channel,
'Use `!check` to check the status, and set them with `!set <param> <value>`.'
)
await self.bot.send_message(
ctx.message.channel,
'If using emojis not in this server, use the fully qualified name, eg `<:downvote:335141916989456384>` while setting `up` and `down`.'
)
def __parse_ticket_from_issue(issue, cloudBase):
# Parse Reporter
reporter = User(issue['fields']['reporter']['displayName'])
# Parse Assignee
assignee = User("Not Assigned")
try:
assignee.username = issue['fields']['assignee']['displayName']
except:
pass
# Parse Key
key = issue['key']
link = JIRA_USER_TICKET.replace(JIRA_USER_BASE_REPLACE, cloudBase)
link = link.replace(JIRA_USER_TICKET_REPLACE, key)
# Parse Id
issueId = issue['id']
# Parse Summary
summary = issue['fields']['summary']
# Parse created At
createdAt = issue['fields']['created']
createdAt = datetime.datetime.strptime(createdAt, '%Y-%m-%dT%H:%M:%S.%f%z')
# Parse Labels
labels = issue['fields']['labels']
# Parse Status
status = Status(issue['fields']['status']['statusCategory']['name'],
issue['fields']['status']['statusCategory']['colorName'])
# Parse subtasks out
subtasks = []
for subtask in issue['fields']['subtasks']:
subtasks.append(Subtask(subtask['key']))
# Parse type
ticket_type = issue['fields']['issuetype']['name']
ticket = Ticket(issueId, key, summary, key, labels, createdAt, reporter,
assignee, subtasks, link, status, ticket_type)
return ticket
async def add_status(req):
new_status = Status(title=req.json.get("title"), text=req.json.get("text"))
new_status.save()
return json({}, status=201)
