Merge pull request #12 from saisyam/line-number-support

Line number support added

Author: saisyam

gitparse.py

@@ -12,10 +12,38 @@
 
 HEADER = {'User-Agent': random.choice(HEADERS_LIST)}
 
-def parsegit(url):
-    #url = "https://github.com/saisyam/python-flask/blob/master/simple/app.py" 
+def parsegit(url, lineno): 
     response = requests.get(url, headers=HEADER)
     soup = BeautifulSoup(response.text, "lxml")
     div = soup.find_all("div", {"itemprop" : "text"})
     div[0].find('details').decompose()
-    return str(div[0])
+    
+    if lineno is not None:
+        trs = div[0].find_all('tr')
+        # split line by comma separated
+        # 1,5 - display 1 and 5 lines
+        # 1,3-5,8 - display 1,3,4,5 and 8 lines
+        vlines = linestodisplay(lineno, len(trs))
+        index = 1
+        for tr in trs:
+            if index not in vlines:
+                tr.decompose()
+            index = index + 1
+
+    return str(div[0])
+
+def linestodisplay(linestring, maxlines):
+    lines = linestring.split(",")
+    vlines = []
+    for line in lines:
+        #find if there is any range
+        rlines = line.split("-")
+        if len(rlines) > 1:
+            vlines.extend(range(int(rlines[0]), int(rlines[1])))
+            vlines.append(int(rlines[1]))
+        else:
+            vlines.append(int(rlines[0]))
+    vlines.sort()
+    while vlines[-1] > int(maxlines):
+        vlines.remove(vlines[-1])
+    return vlines

main.py

@@ -11,12 +11,29 @@
 def index():
     return "<h1>Landing page coming soon...</h1>"
 
+# provide github URL to get complete code
+# to give linenumbers add /l/<lineno> - 1,2 or 1,3-5,8 etc
 @app.route('/<path:url>')
 def gistify(url):
-    filename = url.split("/")[-1]
-    rawurl = url.replace("blob", "raw")
-    document = parsegit(url)
-    return render_template('gistify.html', document=Markup(document), filename=filename, rawurl=rawurl, url=url)
+    try:
+        urlitems = url.split("/")
+        filename = None
+        lineno = None
+        giturl = None
+        
+        if urlitems[-2] == "l":
+            lineno = urlitems[-1]
+            filename = urlitems[-3]
+            giturl = url[:url.index("/l/")]
+        else:
+            filename = urlitems[-1]
+            lineno = None
+            giturl = url
+        rawurl = giturl.replace("blob", "raw")
+        document = parsegit(giturl, lineno)
+        return render_template('gistify.html', document=Markup(document), filename=filename, rawurl=rawurl, url=giturl)
+    except:
+        return render_template('error.html', url=url)
 
 if __name__ == '__main__':
     app.run(host='127.0.0.1', port=8080, debug=True)

templates/error.html

@@ -0,0 +1,4 @@
+<h2>Oops!</h2>
+<p>Something went wrong with the URL- <a href="{{ url }}">{{ url }}</a>. <br \>
+    Please check the guide at <a href="http://gistify.saisyam.com">http://gistify.saisyam.com</a>
+</p>