+ """star_a_project

+ 

+ Revision ID: c34f4b09ef18

+ Revises: 8a5d68f74beb

+ Create Date: 2017-07-07 00:08:18.257075

+ 

+ """

+ 

+ # revision identifiers, used by Alembic.

+ revision = 'c34f4b09ef18'

+ down_revision = '8a5d68f74beb'

+ 

+ from alembic import op

+ import sqlalchemy as sa

+ 

+ 

+ def upgrade():

+     ''' Add a new table to store data about who starred which project '''

+     op.create_table(

+         'stargazers',

+         sa.Column('id', sa.Integer, primary_key=True),

+         sa.Column(

+             'project_id',

+             sa.Integer,

+             sa.ForeignKey(

+                 'projects.id', onupdate='CASCADE', ondelete='CASCADE'),

+             nullable=False,

+             index=True,

+         ),

+         sa.Column(

+             'user_id',

+             sa.Integer,

+             sa.ForeignKey('users.id', onupdate='CASCADE', ondelete='CASCADE'),

+             nullable=False,

+         )

+     )

+ 

+     op.create_unique_constraint(

+         constraint_name='uq_stargazers_project_id_user_id_key',

+         table_name='stargazers',

+         columns=['project_id', 'user_id']

+     )

+ 

+ 

+ def downgrade():

+     ''' Remove the stargazers table from the database '''

+     op.drop_constraint(

+         constraint_name='uq_stargazers_project_id_user_id_key',

+         table_name='stargazers'

+     )

+     op.drop_table('stargazers')

+             flask.g.repo_starred = pagure.lib.has_starred(

+                 SESSION, flask.g.repo, user=flask.g.fas_user.username,

+             )

+ 

+ 

+ def update_star_project(session, repo, star, user):

+     ''' Unset or set the star status depending on the star value.

+ 

+     :arg session: the session to use to connect to the database.

+     :arg repo: a model.Project object representing the project to star/unstar

+     :arg star: '1' for starring and '0' for unstarring

+     :arg user: string representing the user

+     :return: None or string containing 'You starred this project' or

+             'You unstarred this project'

+     '''

+ 

+     if not all([repo, user, star]):

+         return

+     user_obj = get_user(session, user)

+     msg = None

+     if star == '1':

+         msg = _star_project(

+             session,

+             repo=repo,

+             user=user_obj,

+         )

+     elif star == '0':

+         msg = _unstar_project(

+             session,

+             repo=repo,

+             user=user_obj,

+         )

+     return msg

+ 

+ 

+ def _star_project(session, repo, user):

+     ''' Star a project

+ 

+     :arg session: Session object to connect to db with

+     :arg repo: model.Project object representing the repo to star

+     :arg user: model.User object who is starring this repo

+     :return: None or string containing 'You starred this project'

+     '''

+ 

+     if not all([repo, user]):

+         return

+     stargazer_obj = model.Star(

+         project_id=repo.id,

+         user_id=user.id,

+     )

+     session.add(stargazer_obj)

+     return 'You starred this project'

+ 

+ 

+ def _unstar_project(session, repo, user):

+     ''' Unstar a project

+     :arg session: Session object to connect to db with

+     :arg repo: model.Project object representing the repo to unstar

+     :arg user: model.User object who is unstarring this repo

+     :return: None or string containing 'You unstarred this project'

+             or 'You never starred the project'

+     '''

+ 

+     if not all([repo, user]):

+         return

+     # First find the stargazer_obj object

+     stargazer_obj = _get_stargazer_obj(session, repo, user)

+     if isinstance(stargazer_obj, model.Star):

+         session.delete(stargazer_obj)

+         msg = 'You unstarred this project'

+     else:

+         msg = 'You never starred the project'

+     return msg

+ 

+ 

+ def _get_stargazer_obj(session, repo, user):

+     ''' Query the db to find stargazer object with given repo and user

+     :arg session: Session object to connect to db with

+     :arg repo: model.Project object

+     :arg user: model.User object

+     :return: None or model.Star object

+     '''

+ 

+     if not all([repo, user]):

+         return

+     stargazer_obj = session.query(

+         model.Star,

+     ).filter(

+         model.Star.project_id == repo.id,

+     ).filter(

+         model.Star.user_id == user.id,

+     )

+ 

+     return stargazer_obj.first()

+ 

+ 

+ def has_starred(session, repo, user):

+     ''' Check if a given user has starred a particular project

+ 

+     :arg session: The session object to query the db with

+     :arg repo: model.Project object for which the star is checked

+     :arg user: The username of the user in question

+     :return: True if user has starred the project, False otherwise

+     '''

+ 

+     if not all([repo, user]):

+         return

+     user_obj = search_user(session, username=user)

+     stargazer_obj = _get_stargazer_obj(session, repo, user_obj)

+     if isinstance(stargazer_obj, model.Star):

+         return True

+     return False

+ class Star(BASE):

+     """ Stores users association with the all the projects which

+     they have starred

+ 

+     Table -- star

+     """

+ 

+     __tablename__ = 'stargazers'

+     __table_args__ = (

+         sa.UniqueConstraint(

+             'project_id',

+             'user_id',

+             name='uq_stargazers_project_id_user_id_key'),

+     )

+ 

+     id = sa.Column(sa.Integer, primary_key=True)

+     project_id = sa.Column(

+         sa.Integer,

+         sa.ForeignKey('projects.id', onupdate='CASCADE', ondelete='CASCADE'),

+         nullable=False,

+         index=True,

+     )

+     user_id = sa.Column(

+         sa.Integer,

+         sa.ForeignKey('users.id', onupdate='CASCADE', ondelete='CASCADE'),

+         nullable=False,

+     )

+     user = relation(

+         'User', foreign_keys=[user_id], remote_side=[User.id],

+         backref=backref(

+             'stars', cascade="delete, delete-orphan"

+         ),

+     )

+     project = relation(

+         'Project', foreign_keys=[project_id], remote_side=[Project.id],

+         backref=backref(

+             'stargazers', cascade="delete, delete-orphan",

+         ),

+     )

+ 

+ 

+                   <a class="dropdown-item" href="{{

+                     url_for('view_user_stars', username=g.fas_user.username)

+                     }}">My Stars</a>

+                   <div class="dropdown-divider"></div>

+ 

+           {% if not g.repo_starred %}

+             <form method="POST" name="star_project" id="star_project"

+               action="{{ url_for(

+               '.star_project',

+               repo=repo.name,

+               username=username,

+               namespace=repo.namespace,

+               star=1)}}">

+               {{ forkbuttonform.csrf_token }}

+             </form>

+               <button class="btn btn-sm btn-secondary"

+               onclick="$('#star_project').submit();">

+                 <span class="oi" data-glyph="star"></span>Star</button>

+                 <a href="{{url_for(

+                             'view_stargazers',

+                             repo=repo.name,

+                             username=repo.user.user if repo.is_fork else None,

+                             namespace=repo.namespace,

+                 )}}" class="btn btn-sm btn-primary">{{repo.stargazers|length}}</a>

+           {% else %}

+             <form method="POST" name="unstar_project" id="unstar_project"

+               action="{{ url_for(

+               '.star_project',

+               repo=repo.name,

+               username=username,

+               namespace=repo.namespace,

+               star=0)}}">

+               {{ forkbuttonform.csrf_token }}

+             </form>

+               <button class="btn btn-sm btn-secondary"

+               onclick="$('#unstar_project').submit();">

+                 <span class="oi" data-glyph="star"></span>Unstar</button>

+                 <a href="{{url_for(

+                             'view_stargazers',

+                             repo=repo.name,

+                             username=repo.user.user if repo.is_fork else None,

+                             namespace=repo.namespace,

+                 )}}" class="btn btn-sm btn-primary">{{repo.stargazers|length}}</a>

+           {% endif %}

+ 	</div>

+ 

+ 	<div class="btn-group">

+ {% extends "repo_master.html" %}

+ 

+ {% block title %}Stargazers of {{ repo.fullname }} {% endblock %}

+ 

+ {% block content %}

+ <div class="p-t-2">

+   <div class="container">

+     <h3 class="p-b-2">

+       Starred by {{ users | length}} users

+     </h3>

+     <hr>

+       <table class="table table-hover table-striped m-b-0 m-t-1">

+         <thead class="thead-default">

+           <tr>

+             <th class="nowrap">Stargazers of {{ repo.fullname }} </th>

+           </tr>

+         </thead>

+         <tbody>

+           {% for user in users %}

+             <tr class="nowrap">

+               <td>

+                 <img src="{{ user.default_email | avatar_url(36) | safe }}" width=20 height=20 />

+               <a href="{{ url_for(

+                 'view_user',

+                 username=user.user,

+               )}}">{{user.user}}

+               </a></br>

+               </td>

+             </tr>

+           {% endfor %}

+         </tbody>

+       </table>

+       <br>

+   </div>

+ </div>

+ {% endblock %}

+ {% extends "master.html" %}

+ 

+ {% block title %}{{ username }}'s starred Projects{% endblock %}

+ 

+ {% block content %}

+ <div class="p-t-2">

+   <div class="container">

+     <h3 class="p-b-2">

+       {{ repos | length }} Projects Starred by {{ username }}

+     </h3>

+     <hr>

+       <table class="table table-hover table-striped m-b-0 m-t-1">

+         <thead class="thead-default">

+           <tr>

+             <th class="nowrap">Starred Projects</th>

+           </tr>

+         </thead>

+         <tbody>

+           {% for repo in repos %}

+             <tr class="nowrap">

+               <td>

+                 {% if repo.avatar_email %}

+                 <img src="{{ repo.avatar_email | avatar_url | safe }}" width=20 height=20 />

+                 <a href="{{ url_for(

+                 'view_repo',

+                 repo=repo.name,

+                 username=repo.user.user if repo.is_fork else None,

+                 namespace=repo.namespace,

+                 )}}">{{repo.fullname}}</a>

+                 </br>

+                 {% else %}

+                <div class="col-sm-12">

+                  <span class="oi" data-glyph="document"></span>

+                 <a href="{{ url_for(

+                 'view_repo',

+                 repo=repo.name,

+                 username=repo.user.user if repo.is_fork else None,

+                 namespace=repo.namespace,

+                 )}}">{{repo.fullname}}</a>

+                 </div>

+                 </br>

+                 {% endif %}

+               </td>

+             </tr>

+           {% endfor %}

+         </tbody>

+       </table>

+       <br>

+   </div>

+ </div>

+ {% endblock %}

+ @APP.route('/user/<username>/stars/')

+ @APP.route('/user/<username>/stars')

+ def view_user_stars(username):

+     """

+     Shows the starred projects of the specified user.

+ 

+     :arg username: The username whose stars we have to retrieve

+     """

+ 

+     user = _get_user(username=username)

+ 

+     return flask.render_template(

+         'user_stars.html',

+         username=username,

+         user=user,

+         repos=[star.project for star in user.stars],

+     )

+ 

+ 

+ @APP.route('/<repo>/stargazers/')

+ @APP.route('/fork/<username>/<repo>/stargazers/')

+ @APP.route('/<namespace>/<repo>/stargazers/')

+ @APP.route('/fork/<username>/<namespace>/<repo>/stargazers/')

+ def view_stargazers(repo, username=None, namespace=None):

+     ''' View all the users who have starred the project '''

+ 

+     stargazers = flask.g.repo.stargazers

+     users = [star.user for star in stargazers]

+     return flask.render_template(

+         'repo_stargazers.html', repo=flask.g.repo, users=users)

+ 

+ 

+ @APP.route('/<repo>/star/<star>', methods=["POST"])

+ @APP.route('/fork/<username>/<repo>/star/<star>', methods=["POST"])

+ @APP.route('/<namespace>/<repo>/star/<star>', methods=["POST"])

+ @APP.route('/fork/<username>/<namespace>/<repo>/star/<star>', methods=["POST"])

+ @login_required

+ def star_project(repo, star, username=None, namespace=None):

+     ''' Star or Unstar a project

+ 

+     :arg repo: string representing the project which has to be starred or

+     unstarred.

+     :arg star: either '0' or '1' for unstar and star respectively

+     :arg username: string representing the user the fork of whose is being

+     starred or unstarred.

+     :arg namespace: namespace of the project if any

+     '''

+ 

+     return_point = flask.url_for('index')

+     if (

+             flask.request.referrer is not None and

+             pagure.is_safe_url(flask.request.referrer)):

+         return_point = flask.request.referrer

+ 

+     form = pagure.forms.ConfirmationForm()

+     if not form.validate_on_submit():

+         flask.abort(400)

+ 

+     if star not in ['0', '1']:

+         flask.abort(400)

+ 

+     try:

+         msg = pagure.lib.update_star_project(

+             SESSION,

+             user=flask.g.fas_user.username,

+             repo=flask.g.repo,

+             star=star,

+         )

+         SESSION.commit()

+         flask.flash(msg)

+     except SQLAlchemyError:

+         flask.flash('Could not star the project')

+ 

+     return flask.redirect(return_point)

+ 

+ 

+ # coding=utf-8

+ 

+ """

+  (c) 2017 - Copyright Red Hat Inc

+ 

+  Authors:

+    Pierre-Yves Chibon <pingou@pingoured.fr>

+    Vivek Anand <vivekanand1101@gmail.com>

+ 

+ """

+ 

+ __requires__ = ['SQLAlchemy >= 0.8']

+ import pkg_resources

+ 

+ import unittest

+ import sys

+ import os

+ 

+ sys.path.insert(0, os.path.join(os.path.dirname(

+     os.path.abspath(__file__)), '..'))

+ 

+ import pagure

+ import pagure.lib

+ import tests

+ 

+ 

+ class TestStarProjectUI(tests.SimplePagureTest):

+     def setUp(self):

+         """ Set up the environnment, ran before every tests. """

+         super(TestStarProjectUI, self).setUp()

+ 

+         pagure.APP.config['TESTING'] = True

+         pagure.SESSION = self.session

+         pagure.ui.SESSION = self.session

+         pagure.ui.app.SESSION = self.session

+         pagure.ui.filters.SESSION = self.session

+         pagure.ui.repo.SESSION = self.session

+         pagure.ui.issues.SESSION = self.session

+ 

+         tests.create_projects(self.session)

+         tests.create_projects_git(os.path.join(self.path, 'repos'), bare=True)

+ 

+     def _check_star_count(self, data, stars=1):

+         """ Check if the star count is correct or not """

+         output = self.app.get(

+             '/test/', data=data, follow_redirects=True)

+         if stars == 1:

+             self.assertIn(

+                 '<a href="/test/stargazers/" class="btn '

+                 'btn-sm btn-primary">1</a>',

+                 output.data

+             )

+         elif stars == 0:

+             self.assertIn(

+                 '<a href="/test/stargazers/" class="btn '

+                 'btn-sm btn-primary">0</a>',

+                 output.data

+             )

+ 

+     def test_star_project_no_project(self):

+         """ Test the star_project endpoint. """

+ 

+         # No such project

+         output = self.app.post('/test42/star/1')

+         self.assertEqual(output.status_code, 404)

+ 

+     def test_star_project_no_csrf(self):

+         """ Test the star_project endpoint for the case when there

+         is no CSRF token given """

+ 

+         user = tests.FakeUser()

+         user.username = 'pingou'

+         with tests.user_set(pagure.APP, user):

+ 

+             data = {}

+             output = self.app.post(

+                 '/test/star/1', data=data, follow_redirects=True)

+             self.assertEqual(output.status_code, 400)

+ 

+     def test_star_project_invalid_star(self):

+         """ Test the star_project endpoint for invalid star """

+ 

+         user = tests.FakeUser()

+         user.username = 'pingou'

+         with tests.user_set(pagure.APP, user):

+             csrf_token = self.get_csrf()

+             data = {

+                 'csrf_token': csrf_token,

+             }

+             output = self.app.post(

+                 '/test/star/2', data=data, follow_redirects=True)

+             self.assertEqual(output.status_code, 400)

+             self._check_star_count(data=data, stars=0)

+ 

+     def test_star_project_valid_star(self):

+         """ Test the star_project endpoint for correct star """

+ 

+         user = tests.FakeUser()

+         user.username = 'pingou'

+         with tests.user_set(pagure.APP, user):

+             csrf_token = self.get_csrf()

+             data = {

+                 'csrf_token': csrf_token,

+             }

+ 

+             # try starring the project for pingou

+             output = self.app.post(

+                 '/test/star/1', data=data, follow_redirects=True)

+             self.assertEqual(output.status_code, 200)

+             self.assertIn(

+                 '</button>\n                      You starred '

+                 'this project\n                    </div>',

+                 output.data

+             )

+ 

+             # check home page of project for star count

+             self._check_star_count(data=data, stars=1)

+ 

+             # try unstarring the project for pingou

+             output = self.app.post(

+                 '/test/star/0', data=data, follow_redirects=True)

+             self.assertEqual(output.status_code, 200)

+             self.assertIn(

+                 '</button>\n                      You unstarred '

+                 'this project\n                    </div>',

+                 output.data

+             )

+             self._check_star_count(data=data, stars=0)

+ 

+     def test_repo_stargazers(self):

+         """ Test the repo_stargazers endpoint of pagure.ui.repo """

+ 

+         # make pingou star the project

+         # first create pingou

+         user = tests.FakeUser()

+         user.username = 'pingou'

+         with tests.user_set(pagure.APP, user):

+             csrf_token = self.get_csrf()

+             data = {

+                 'csrf_token': csrf_token,

+             }

+ 

+             output = self.app.post(

+                 '/test/star/1', data=data, follow_redirects=True)

+             self.assertEqual(output.status_code, 200)

+             self.assertIn(

+                 '</button>\n                      You starred '

+                 'this project\n                    </div>',

+                 output.data

+             )

+             self._check_star_count(data=data, stars=1)

+ 

+         # now, test if pingou's name comes in repo stargazers

+         output = self.app.get(

+             '/test/stargazers/'

+         )

+         self.assertIn(

+             '<title>Stargazers of test  - Pagure</title>',

+             output.data

+         )

+         self.assertIn(

+             '<a href="/user/pingou">pingou\n              </a>',

+             output.data

+         )

+ 

+         # make pingou unstar the project

+         user = tests.FakeUser()

+         user.username = 'pingou'

+         with tests.user_set(pagure.APP, user):

+             csrf_token = self.get_csrf()

+             data = {

+                 'csrf_token': csrf_token,

+             }

+ 

+             output = self.app.post(

+                 '/test/star/0', data=data, follow_redirects=True)

+             self.assertEqual(output.status_code, 200)

+             self.assertIn(

+                 '</button>\n                      You unstarred '

+                 'this project\n                    </div>',

+                 output.data

+             )

+             self._check_star_count(data=data, stars=0)

+ 

+         # now, test if pingou's name comes in repo stargazers

+         # it shouldn't because, he just unstarred

+         output = self.app.get(

+             '/test/stargazers/'

+         )

+         self.assertIn(

+             '<title>Stargazers of test  - Pagure</title>',

+             output.data

+         )

+         self.assertNotIn(

+             '<a href="/user/pingou">pingou\n              </a>',

+             output.data

+         )

+ 

+     def test_user_stars(self):

+         """ Test the user_stars endpoint of pagure.ui.app """

+ 

+         # make pingou star the project

+         # first create pingou

+         user = tests.FakeUser()

+         user.username = 'pingou'

+         with tests.user_set(pagure.APP, user):

+             csrf_token = self.get_csrf()

+             data = {

+                 'csrf_token': csrf_token,

+             }

+ 

+             output = self.app.post(

+                 '/test/star/1', data=data, follow_redirects=True)

+             self.assertEqual(output.status_code, 200)

+             self.assertIn(

+                 '</button>\n                      You starred '

+                 'this project\n                    </div>',

+                 output.data

+             )

+             self._check_star_count(data=data, stars=1)

+ 

+         # now, test if the project 'test' comes in pingou's stars

+         output = self.app.get(

+             '/user/pingou/stars'

+         )

+         self.assertIn(

+             "<title>pingou's starred Projects - Pagure</title>",

+             output.data

+         )

+         self.assertIn(

+             '<a href="/test">test</a>\n',

+             output.data

+         )

+ 

+         # make pingou unstar the project

+         user = tests.FakeUser()

+         user.username = 'pingou'

+         with tests.user_set(pagure.APP, user):

+             csrf_token = self.get_csrf()

+             data = {

+                 'csrf_token': csrf_token,

+             }

+             output = self.app.post(

+                 '/test/star/0', data=data, follow_redirects=True)

+             self.assertEqual(output.status_code, 200)

+             self.assertIn(

+                 '</button>\n                      You unstarred '

+                 'this project\n                    </div>',

+                 output.data

+             )

+             self._check_star_count(data=data, stars=0)

+ 

+         # now, test if test's name comes in pingou's stars

+         # it shouldn't because, he just unstarred

+         output = self.app.get(

+             '/user/pingou/stars/'

+         )

+         self.assertIn(

+             "<title>pingou's starred Projects - Pagure</title>",

+             output.data

+         )

+         self.assertNotIn(

+             '<a href="/test">test</a>\n',

+             output.data

+         )

+ 

+ 

+ if __name__ == '__main__':

+     unittest.main(verbosity=2)

+ # coding=utf-8

+ """

+  (c) 2015-2017 - Copyright Red Hat Inc

+ 

+  Authors:

+    Pierre-Yves Chibon <pingou@pingoured.fr>

+    Vivek Anand <vivekanand1101@gmail.com>

+ 

+ """

+ 

+ __requires__ = ['SQLAlchemy >= 0.8']

+ import pkg_resources

+ 

+ import unittest

+ import sys

+ import os

+ 

+ sys.path.insert(0, os.path.join(os.path.dirname(

+     os.path.abspath(__file__)), '..'))

+ 

+ import pagure

+ import pagure.lib

+ import tests

+ 

+ 

+ class TestStarProjectLib(tests.SimplePagureTest):

+     ''' Test the star project feature of pagure '''

+ 

+     def setUp(self):

+         """ Set up the environnment for running each star project lib tests """

+         super(TestStarProjectLib, self).setUp()

+         tests.create_projects(self.session)

+ 

+     def test_update_star_project(self):

+         ''' Test the update_star_project endpoint of pagure.lib '''

+ 

+         repo_obj = pagure.lib._get_project(self.session, 'test')

+         # test with invalud Star object, should return None

+         msg = pagure.lib.update_star_project(

+             self.session,

+             repo_obj,

+             None,

+             'pingou',

+         )

+         self.session.commit()

+         self.assertEqual(msg, None)

+         user_obj = pagure.lib.get_user(self.session, 'pingou')

+         self.assertEqual(len(user_obj.stars), 0)

+ 

+         # test starring the project

+         repo_obj = pagure.lib._get_project(self.session, 'test')

+         msg = pagure.lib.update_star_project(

+             self.session,

+             repo_obj,

+             '1',

+             'pingou',

+         )

+ 

+         self.session.commit()

+         self.assertEqual(msg, 'You starred this project')

+         user_obj = pagure.lib.get_user(self.session, 'pingou')

+         self.assertEqual(len(user_obj.stars), 1)

+ 

+         # test unstarring the project

+         repo_obj = pagure.lib._get_project(self.session, 'test')

+         msg = pagure.lib.update_star_project(

+             self.session,

+             repo_obj,

+             '0',

+             'pingou',

+         )

+ 

+         self.session.commit()

+         self.assertEqual(msg, 'You unstarred this project')

+         user_obj = pagure.lib.get_user(self.session, 'pingou')

+         self.assertEqual(len(user_obj.stars), 0)

+ 

+     def test_star_project(self):

+         ''' Test the _star_project endpoint of pagure.lib '''

+ 

+         # test with not all arguments present

+         user_obj = pagure.lib.get_user(self.session, 'pingou')

+         repo_obj = pagure.lib._get_project(self.session, 'test')