v2_app.py

import traceback,re
from flask import Blueprint, jsonify, request
import markdown
from utils import require_secret_key
from utils import determine_week
from v2_utils import calculate_overall_progress, define_link_data, week_data_formatter
# from query import PostgresORM
from shared_migrations.db.dmp_api import DmpAPIQueries
# from app import async_session
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
from sqlalchemy.orm import sessionmaker
from sqlalchemy.pool import NullPool
from shared_migrations.db import get_postgres_uri


v2 = Blueprint('v2', __name__)


engine = create_async_engine(get_postgres_uri(), echo=False,poolclass=NullPool)
async_session = sessionmaker(autocommit=False, autoflush=False, bind=engine, class_=AsyncSession)

@v2.route('/issues/<owner>/<issue>', methods=['GET'])
# @require_secret_key
async def get_issues_by_owner_id_v2(owner, issue):
    
    try:                 
        # Fetch issue updates based on owner and issue number
        
        url = f"https://github.com/{owner}"        
        
        # import pdb;pdb.set_trace()
        actual_owner = await DmpAPIQueries.get_actual_owner_query(async_session, owner)
        repo_owner =actual_owner[0]['repo_owner'] if actual_owner else ""
        #create url with repo owner
        url = f"https://github.com/{repo_owner}" if repo_owner else None
        
        dmp_issue_id = await DmpAPIQueries.get_dmp_issues(async_session, issue)
        if not dmp_issue_id:
          print(f"url....{url}....{issue}")
          return jsonify({'error': "No data found in dmp_issue"}), 500
        
        dmp_issue_id = dmp_issue_id[0]        

        response = await DmpAPIQueries.get_dmp_issue_updates(async_session, dmp_issue_id['id'])    
        if not response:
            print(f"dmp_issue_id....{response}....{dmp_issue_id}")
            return jsonify({'error': "No data found in dmp_issue_updates"}), 500

        data = response
        
        final_data = []
        w_learn_url,w_goal_url,avg,cont_details,plain_text_body,plain_text_wurl = None,None,None,None,None,None

        
        for val in data:
            # issue_url = "https://api.github.com/repos/{}/{}/issues/comments".format(val['owner'],val['repo'])
            # week_avg ,cont_name,cont_id,w_goal,w_learn,weekby_avgs,org_link = find_week_avg(issue_url)
            # mentors_data = find_mentors(val['issue_url']) if val['issue_url'] else {'mentors': [], 'mentor_usernames': []}
            
            if val['body_text']:                                
                if ("Weekly Goals" in val['body_text'] and not w_goal_url):
                    w_goal_url = val['body_text']
                    plain_text_body = markdown.markdown(val['body_text'])
                    tasks = re.findall(r'\[(x| )\]', plain_text_body)
                    total_tasks = len(tasks)
                    completed_tasks = tasks.count('x')
                    avg = round((completed_tasks/total_tasks)*100) if total_tasks!=0 else 0

                if ("Weekly Learnings" in val['body_text'] and not w_learn_url):
                    w_learn_url = val['body_text']
                    plain_text_wurl = markdown.markdown(val['body_text'])

            
            # mentors = mentors_data['mentors']
            # ment_usernames = mentors_data['mentor_usernames']
            if not cont_details:
                cont_details = dmp_issue_id['contributor_username']
        week_data = week_data_formatter(plain_text_body,"Goals")
        res = {
            "name": dmp_issue_id['title'],
            "description": dmp_issue_id['description'],
            "mentor": define_link_data(dmp_issue_id['mentor_username']),
            "mentor_id": dmp_issue_id['mentor_username'] ,
            "contributor":define_link_data(cont_details),
            # "contributor_id": cont_details[0]['contributor_id'],
            "org": define_link_data(repo_owner)[0] if repo_owner else None,
            "weekly_goals_html": w_goal_url,
            "weekly_learnings_html": w_learn_url,
            "overall_progress":calculate_overall_progress(week_data,12),
            "issue_url":dmp_issue_id['issue_url'],
            "pr_details":None,
            "weekly_goals":week_data,
            "weekly_learnings":week_data_formatter(plain_text_wurl,"Learnings")
        }
        
        
        pr_Data = await DmpAPIQueries.get_pr_data(async_session, dmp_issue_id['id'])        
        transformed = {"pr_details": []}
        if pr_Data:
            for pr in pr_Data:
                pr_status = pr.get("status", "")
                if pr_status == "closed" and pr.get("merged_at"):
                    pr_status = "merged"
                transformed["pr_details"].append({
                    "id": pr.get("pr_id", ""),
                    "name": pr.get("title", ""),
                    "week": determine_week(pr['created_at']),
                    "link": pr.get("link", ""),
                    "status": pr_status ,
                })
                
        res['pr_details'] = transformed['pr_details']
                
        return jsonify(res),200
  
    except Exception as e:
        error_traceback = traceback.format_exc()
        return jsonify({'error': str(e), 'traceback': error_traceback}), 200