summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--aurweb/exceptions.py12
-rwxr-xr-xaurweb/git/serve.py63
2 files changed, 75 insertions, 0 deletions
diff --git a/aurweb/exceptions.py b/aurweb/exceptions.py
index 5922b2df..639f9e09 100644
--- a/aurweb/exceptions.py
+++ b/aurweb/exceptions.py
@@ -48,6 +48,18 @@ class InvalidCommentException(AurwebException):
super(InvalidCommentException, self).__init__(msg)
+class AlreadyVotedException(AurwebException):
+ def __init__(self, comment):
+ msg = 'already voted for package base: {:s}'.format(comment)
+ super(AlreadyVotedException, self).__init__(msg)
+
+
+class NotVotedException(AurwebException):
+ def __init__(self, comment):
+ msg = 'missing vote for package base: {:s}'.format(comment)
+ super(NotVotedException, self).__init__(msg)
+
+
class InvalidArgumentsException(AurwebException):
def __init__(self, msg):
super(InvalidArgumentsException, self).__init__(msg)
diff --git a/aurweb/git/serve.py b/aurweb/git/serve.py
index e33fbeb4..4c03e3b6 100755
--- a/aurweb/git/serve.py
+++ b/aurweb/git/serve.py
@@ -318,6 +318,57 @@ def pkgbase_unflag(pkgbase, user):
conn.commit()
+def pkgbase_vote(pkgbase, user):
+ pkgbase_id = pkgbase_from_name(pkgbase)
+ if not pkgbase_id:
+ raise aurweb.exceptions.InvalidPackageBaseException(pkgbase)
+
+ conn = aurweb.db.Connection()
+
+ cur = conn.execute("SELECT ID FROM Users WHERE Username = ?", [user])
+ userid = cur.fetchone()[0]
+ if userid == 0:
+ raise aurweb.exceptions.InvalidUserException(user)
+
+ cur = conn.execute("SELECT COUNT(*) FROM PackageVotes " +
+ "WHERE UsersID = ? AND PackageBaseID = ?",
+ [userid, pkgbase_id])
+ if cur.fetchone()[0] > 0:
+ raise aurweb.exceptions.AlreadyVotedException(pkgbase)
+
+ now = int(time.time())
+ conn.execute("INSERT INTO PackageVotes (UsersID, PackageBaseID, VoteTS) " +
+ "VALUES (?, ?, ?)", [userid, pkgbase_id, now])
+ conn.execute("UPDATE PackageBases SET NumVotes = NumVotes + 1 " +
+ "WHERE ID = ?", [pkgbase_id])
+ conn.commit()
+
+
+def pkgbase_unvote(pkgbase, user):
+ pkgbase_id = pkgbase_from_name(pkgbase)
+ if not pkgbase_id:
+ raise aurweb.exceptions.InvalidPackageBaseException(pkgbase)
+
+ conn = aurweb.db.Connection()
+
+ cur = conn.execute("SELECT ID FROM Users WHERE Username = ?", [user])
+ userid = cur.fetchone()[0]
+ if userid == 0:
+ raise aurweb.exceptions.InvalidUserException(user)
+
+ cur = conn.execute("SELECT COUNT(*) FROM PackageVotes " +
+ "WHERE UsersID = ? AND PackageBaseID = ?",
+ [userid, pkgbase_id])
+ if cur.fetchone()[0] == 0:
+ raise aurweb.exceptions.NotVotedException(pkgbase)
+
+ conn.execute("DELETE FROM PackageVotes WHERE UsersID = ? AND " +
+ "PackageBaseID = ?", [userid, pkgbase_id])
+ conn.execute("UPDATE PackageBases SET NumVotes = NumVotes - 1 " +
+ "WHERE ID = ?", [pkgbase_id])
+ conn.commit()
+
+
def pkgbase_set_keywords(pkgbase, keywords):
pkgbase_id = pkgbase_from_name(pkgbase)
if not pkgbase_id:
@@ -467,6 +518,16 @@ def serve(action, cmdargv, user, privileged, remote_addr):
pkgbase = cmdargv[1]
pkgbase_unflag(pkgbase, user)
+ elif action == 'vote':
+ checkarg(cmdargv, 'repository name')
+
+ pkgbase = cmdargv[1]
+ pkgbase_vote(pkgbase, user)
+ elif action == 'unvote':
+ checkarg(cmdargv, 'repository name')
+
+ pkgbase = cmdargv[1]
+ pkgbase_unvote(pkgbase, user)
elif action == 'set-comaintainers':
checkarg_atleast(cmdargv, 'repository name')
@@ -485,6 +546,8 @@ def serve(action, cmdargv, user, privileged, remote_addr):
"set-keywords <name> [...]": "Change package base keywords.",
"setup-repo <name>": "Create a repository (deprecated).",
"unflag <name>": "Remove out-of-date flag from a package base.",
+ "unvote <name>": "Remove vote from a package base.",
+ "vote <name>": "Vote for a package base.",
"git-receive-pack": "Internal command used with Git.",
"git-upload-pack": "Internal command used with Git.",
}