using matplotlib to plot

jaggerwang · jaggerwang · commit 1d4f09f6d332 · 2018-08-31T12:04:10.000+08:00
diff --git a/seiya/seiya/web/app.py b/seiya/seiya/web/app.py
@@ -1,4 +1,4 @@
-from flask import Flask, render_template, jsonify
+from flask import Flask, render_template, jsonify, Response
 
 import seiya.web.job as job
 
@@ -50,6 +50,11 @@ def job_hot_tags_json():
     return jsonify(job.hot_tags())
 
 
+@app.route('/job/hot-tags.png')
+def job_hot_tags_plot():
+    return Response(job.hot_tags_plot(format='png'), content_type='image/png')
+
+
 @app.route('/job/experience-stat')
 def job_experience_stat():
     return render_template('job/experience-stat.html', rows=job.experience_stat())
diff --git a/seiya/seiya/web/job.py b/seiya/seiya/web/job.py
@@ -1,5 +1,9 @@
+from io import BytesIO
+
 from sqlalchemy import func, Float, select, and_
 import pandas as pd
+import matplotlib as mpl
+import matplotlib.pyplot as plt
 
 from seiya.db import engine, Session, JobModel
 
@@ -24,7 +28,7 @@ def salary_top10():
     return [row._asdict() for row in rows]
 
 
-def hot_tags():
+def _hot_tags():
     df = pd.read_sql(select([JobModel.tags]), engine)
 
     df = pd.concat([pd.Series(row['tags'].split(' '))
@@ -35,15 +39,32 @@ def hot_tags():
     df = df[df['tag'] != '""']
     df = df[df['tag'] != '']
 
-    s = df.groupby(['tag']).size().sort_values(ascending=False)
+    return df.groupby(['tag']).size().sort_values(ascending=False)
+
 
+def hot_tags():
     rows = []
-    for item in s.items():
+    for item in _hot_tags().items():
         rows.append({'tag': item[0], 'count': item[1]})
 
     return rows
 
 
+def hot_tags_plot(format='png'):
+    mpl.rcParams['font.sans-serif'] = ['SimHei']
+    mpl.rcParams['axes.unicode_minus'] = False
+    mpl.rcParams['figure.figsize'] = 10, 5
+
+    s = _hot_tags()
+
+    plt.bar(s.index[:10], s.values[:10])
+
+    img = BytesIO()
+    plt.savefig(img, format=format)
+
+    return img.getvalue()
+
+
 def experience_stat():
     session = Session()
     rows = session.query(
diff --git a/seiya/seiya/web/templates/job/hot-tags.html b/seiya/seiya/web/templates/job/hot-tags.html
@@ -11,7 +11,11 @@
     </ol>
 </nav>
 
-<div class="my-5" id="chart"></div>
+<div class="text-center my-5">
+    <img src="/job/hot-tags.png" >
+</div>
+
+<!-- <div class="my-5" id="chart"></div> -->
 
 <table class="table table-striped my-5">
     <thead>
@@ -35,17 +39,17 @@
 
 {% block js %}
 <script>
-$.getJSON('/job/hot-tags.json', function (data) {
-    var chart = new G2.Chart({
-        container: 'chart',
-        forceFit: true,
-        height: 500
-    });
+// $.getJSON('/job/hot-tags.json', function (data) {
+//     var chart = new G2.Chart({
+//         container: 'chart',
+//         forceFit: true,
+//         height: 500
+//     });
 
-    chart.source(data.slice(0, 10));
-    chart.interval().position('tag*count');
+//     chart.source(data.slice(0, 10));
+//     chart.interval().position('tag*count');
     
-    chart.render();
-});
+//     chart.render();
+// });
 </script>
 {% endblock %}
