You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
314 lines
38 KiB
314 lines
38 KiB
|
|
|
|
|
|
<!DOCTYPE html>
|
|
<html>
|
|
<head>
|
|
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
|
|
|
|
|
|
<meta http-equiv="X-UA-Compatible" content="IE=emulateIE7" />
|
|
<title>Coverage for stock_predict\views.py: 99%</title>
|
|
<link rel="stylesheet" href="style.css" type="text/css">
|
|
|
|
<script type="text/javascript" src="jquery.min.js"></script>
|
|
<script type="text/javascript" src="jquery.hotkeys.js"></script>
|
|
<script type="text/javascript" src="jquery.isonscreen.js"></script>
|
|
<script type="text/javascript" src="coverage_html.js"></script>
|
|
<script type="text/javascript">
|
|
jQuery(document).ready(coverage.pyfile_ready);
|
|
</script>
|
|
</head>
|
|
<body class="pyfile">
|
|
|
|
<div id="header">
|
|
<div class="content">
|
|
<h1>Coverage for <b>stock_predict\views.py</b> :
|
|
<span class="pc_cov">99%</span>
|
|
</h1>
|
|
|
|
<img id="keyboard_icon" src="keybd_closed.png" alt="Show keyboard shortcuts" />
|
|
|
|
<h2 class="stats">
|
|
73 statements
|
|
<span class="run hide_run shortkey_r button_toggle_run">72 run</span>
|
|
<span class="mis shortkey_m button_toggle_mis">1 missing</span>
|
|
<span class="exc shortkey_x button_toggle_exc">0 excluded</span>
|
|
|
|
|
|
</h2>
|
|
</div>
|
|
</div>
|
|
|
|
<div class="help_panel">
|
|
<img id="panel_icon" src="keybd_open.png" alt="Hide keyboard shortcuts" />
|
|
<p class="legend">Hot-keys on this page</p>
|
|
<div>
|
|
<p class="keyhelp">
|
|
<span class="key">r</span>
|
|
<span class="key">m</span>
|
|
<span class="key">x</span>
|
|
<span class="key">p</span> toggle line displays
|
|
</p>
|
|
<p class="keyhelp">
|
|
<span class="key">j</span>
|
|
<span class="key">k</span> next/prev highlighted chunk
|
|
</p>
|
|
<p class="keyhelp">
|
|
<span class="key">0</span> (zero) top of page
|
|
</p>
|
|
<p class="keyhelp">
|
|
<span class="key">1</span> (one) first highlighted chunk
|
|
</p>
|
|
</div>
|
|
</div>
|
|
|
|
<div id="source">
|
|
<table>
|
|
<tr>
|
|
<td class="linenos">
|
|
<p id="n1" class="stm run hide_run"><a href="#n1">1</a></p>
|
|
<p id="n2" class="stm run hide_run"><a href="#n2">2</a></p>
|
|
<p id="n3" class="pln"><a href="#n3">3</a></p>
|
|
<p id="n4" class="stm run hide_run"><a href="#n4">4</a></p>
|
|
<p id="n5" class="stm run hide_run"><a href="#n5">5</a></p>
|
|
<p id="n6" class="stm run hide_run"><a href="#n6">6</a></p>
|
|
<p id="n7" class="stm run hide_run"><a href="#n7">7</a></p>
|
|
<p id="n8" class="stm run hide_run"><a href="#n8">8</a></p>
|
|
<p id="n9" class="stm run hide_run"><a href="#n9">9</a></p>
|
|
<p id="n10" class="stm run hide_run"><a href="#n10">10</a></p>
|
|
<p id="n11" class="pln"><a href="#n11">11</a></p>
|
|
<p id="n12" class="stm run hide_run"><a href="#n12">12</a></p>
|
|
<p id="n13" class="pln"><a href="#n13">13</a></p>
|
|
<p id="n14" class="pln"><a href="#n14">14</a></p>
|
|
<p id="n15" class="pln"><a href="#n15">15</a></p>
|
|
<p id="n16" class="stm run hide_run"><a href="#n16">16</a></p>
|
|
<p id="n17" class="stm run hide_run"><a href="#n17">17</a></p>
|
|
<p id="n18" class="pln"><a href="#n18">18</a></p>
|
|
<p id="n19" class="stm run hide_run"><a href="#n19">19</a></p>
|
|
<p id="n20" class="pln"><a href="#n20">20</a></p>
|
|
<p id="n21" class="stm run hide_run"><a href="#n21">21</a></p>
|
|
<p id="n22" class="stm run hide_run"><a href="#n22">22</a></p>
|
|
<p id="n23" class="stm run hide_run"><a href="#n23">23</a></p>
|
|
<p id="n24" class="stm run hide_run"><a href="#n24">24</a></p>
|
|
<p id="n25" class="stm run hide_run"><a href="#n25">25</a></p>
|
|
<p id="n26" class="stm run hide_run"><a href="#n26">26</a></p>
|
|
<p id="n27" class="pln"><a href="#n27">27</a></p>
|
|
<p id="n28" class="stm run hide_run"><a href="#n28">28</a></p>
|
|
<p id="n29" class="stm run hide_run"><a href="#n29">29</a></p>
|
|
<p id="n30" class="stm run hide_run"><a href="#n30">30</a></p>
|
|
<p id="n31" class="stm run hide_run"><a href="#n31">31</a></p>
|
|
<p id="n32" class="stm run hide_run"><a href="#n32">32</a></p>
|
|
<p id="n33" class="stm run hide_run"><a href="#n33">33</a></p>
|
|
<p id="n34" class="stm run hide_run"><a href="#n34">34</a></p>
|
|
<p id="n35" class="stm run hide_run"><a href="#n35">35</a></p>
|
|
<p id="n36" class="pln"><a href="#n36">36</a></p>
|
|
<p id="n37" class="stm run hide_run"><a href="#n37">37</a></p>
|
|
<p id="n38" class="stm run hide_run"><a href="#n38">38</a></p>
|
|
<p id="n39" class="pln"><a href="#n39">39</a></p>
|
|
<p id="n40" class="stm run hide_run"><a href="#n40">40</a></p>
|
|
<p id="n41" class="stm run hide_run"><a href="#n41">41</a></p>
|
|
<p id="n42" class="stm run hide_run"><a href="#n42">42</a></p>
|
|
<p id="n43" class="stm run hide_run"><a href="#n43">43</a></p>
|
|
<p id="n44" class="stm run hide_run"><a href="#n44">44</a></p>
|
|
<p id="n45" class="stm run hide_run"><a href="#n45">45</a></p>
|
|
<p id="n46" class="pln"><a href="#n46">46</a></p>
|
|
<p id="n47" class="stm run hide_run"><a href="#n47">47</a></p>
|
|
<p id="n48" class="stm run hide_run"><a href="#n48">48</a></p>
|
|
<p id="n49" class="stm run hide_run"><a href="#n49">49</a></p>
|
|
<p id="n50" class="stm run hide_run"><a href="#n50">50</a></p>
|
|
<p id="n51" class="stm run hide_run"><a href="#n51">51</a></p>
|
|
<p id="n52" class="stm run hide_run"><a href="#n52">52</a></p>
|
|
<p id="n53" class="stm run hide_run"><a href="#n53">53</a></p>
|
|
<p id="n54" class="pln"><a href="#n54">54</a></p>
|
|
<p id="n55" class="stm run hide_run"><a href="#n55">55</a></p>
|
|
<p id="n56" class="stm run hide_run"><a href="#n56">56</a></p>
|
|
<p id="n57" class="pln"><a href="#n57">57</a></p>
|
|
<p id="n58" class="stm run hide_run"><a href="#n58">58</a></p>
|
|
<p id="n59" class="pln"><a href="#n59">59</a></p>
|
|
<p id="n60" class="stm run hide_run"><a href="#n60">60</a></p>
|
|
<p id="n61" class="pln"><a href="#n61">61</a></p>
|
|
<p id="n62" class="pln"><a href="#n62">62</a></p>
|
|
<p id="n63" class="pln"><a href="#n63">63</a></p>
|
|
<p id="n64" class="pln"><a href="#n64">64</a></p>
|
|
<p id="n65" class="stm run hide_run"><a href="#n65">65</a></p>
|
|
<p id="n66" class="stm run hide_run"><a href="#n66">66</a></p>
|
|
<p id="n67" class="stm run hide_run"><a href="#n67">67</a></p>
|
|
<p id="n68" class="stm run hide_run"><a href="#n68">68</a></p>
|
|
<p id="n69" class="stm run hide_run"><a href="#n69">69</a></p>
|
|
<p id="n70" class="stm run hide_run"><a href="#n70">70</a></p>
|
|
<p id="n71" class="stm run hide_run"><a href="#n71">71</a></p>
|
|
<p id="n72" class="stm run hide_run"><a href="#n72">72</a></p>
|
|
<p id="n73" class="stm run hide_run"><a href="#n73">73</a></p>
|
|
<p id="n74" class="pln"><a href="#n74">74</a></p>
|
|
<p id="n75" class="pln"><a href="#n75">75</a></p>
|
|
<p id="n76" class="stm run hide_run"><a href="#n76">76</a></p>
|
|
<p id="n77" class="pln"><a href="#n77">77</a></p>
|
|
<p id="n78" class="pln"><a href="#n78">78</a></p>
|
|
<p id="n79" class="pln"><a href="#n79">79</a></p>
|
|
<p id="n80" class="stm run hide_run"><a href="#n80">80</a></p>
|
|
<p id="n81" class="stm run hide_run"><a href="#n81">81</a></p>
|
|
<p id="n82" class="pln"><a href="#n82">82</a></p>
|
|
<p id="n83" class="stm mis"><a href="#n83">83</a></p>
|
|
<p id="n84" class="pln"><a href="#n84">84</a></p>
|
|
<p id="n85" class="stm run hide_run"><a href="#n85">85</a></p>
|
|
<p id="n86" class="stm run hide_run"><a href="#n86">86</a></p>
|
|
<p id="n87" class="pln"><a href="#n87">87</a></p>
|
|
<p id="n88" class="pln"><a href="#n88">88</a></p>
|
|
<p id="n89" class="stm run hide_run"><a href="#n89">89</a></p>
|
|
<p id="n90" class="stm run hide_run"><a href="#n90">90</a></p>
|
|
<p id="n91" class="stm run hide_run"><a href="#n91">91</a></p>
|
|
<p id="n92" class="stm run hide_run"><a href="#n92">92</a></p>
|
|
<p id="n93" class="stm run hide_run"><a href="#n93">93</a></p>
|
|
<p id="n94" class="pln"><a href="#n94">94</a></p>
|
|
<p id="n95" class="stm run hide_run"><a href="#n95">95</a></p>
|
|
<p id="n96" class="stm run hide_run"><a href="#n96">96</a></p>
|
|
<p id="n97" class="pln"><a href="#n97">97</a></p>
|
|
<p id="n98" class="stm run hide_run"><a href="#n98">98</a></p>
|
|
<p id="n99" class="stm run hide_run"><a href="#n99">99</a></p>
|
|
<p id="n100" class="stm run hide_run"><a href="#n100">100</a></p>
|
|
<p id="n101" class="stm run hide_run"><a href="#n101">101</a></p>
|
|
<p id="n102" class="pln"><a href="#n102">102</a></p>
|
|
<p id="n103" class="stm run hide_run"><a href="#n103">103</a></p>
|
|
<p id="n104" class="pln"><a href="#n104">104</a></p>
|
|
<p id="n105" class="pln"><a href="#n105">105</a></p>
|
|
<p id="n106" class="pln"><a href="#n106">106</a></p>
|
|
<p id="n107" class="pln"><a href="#n107">107</a></p>
|
|
<p id="n108" class="pln"><a href="#n108">108</a></p>
|
|
<p id="n109" class="pln"><a href="#n109">109</a></p>
|
|
<p id="n110" class="pln"><a href="#n110">110</a></p>
|
|
<p id="n111" class="pln"><a href="#n111">111</a></p>
|
|
<p id="n112" class="pln"><a href="#n112">112</a></p>
|
|
|
|
</td>
|
|
<td class="text">
|
|
<p id="t1" class="stm run hide_run"><span class="key">import</span> <span class="nam">json</span><span class="strut"> </span></p>
|
|
<p id="t2" class="stm run hide_run"><span class="key">import</span> <span class="nam">os</span><span class="strut"> </span></p>
|
|
<p id="t3" class="pln"><span class="strut"> </span></p>
|
|
<p id="t4" class="stm run hide_run"><span class="key">from</span> <span class="nam">django</span><span class="op">.</span><span class="nam">http</span> <span class="key">import</span> <span class="nam">HttpResponse</span><span class="strut"> </span></p>
|
|
<p id="t5" class="stm run hide_run"><span class="key">from</span> <span class="nam">django</span><span class="op">.</span><span class="nam">shortcuts</span> <span class="key">import</span> <span class="nam">render</span><span class="op">,</span> <span class="nam">get_object_or_404</span><span class="strut"> </span></p>
|
|
<p id="t6" class="stm run hide_run"><span class="key">from</span> <span class="nam">LSTMPredictStock</span> <span class="key">import</span> <span class="nam">run</span><span class="strut"> </span></p>
|
|
<p id="t7" class="stm run hide_run"><span class="key">from</span> <span class="nam">stock_predict</span> <span class="key">import</span> <span class="nam">models</span><span class="strut"> </span></p>
|
|
<p id="t8" class="stm run hide_run"><span class="key">from</span> <span class="nam">datetime</span> <span class="key">import</span> <span class="nam">datetime</span> <span class="key">as</span> <span class="nam">dt</span><span class="strut"> </span></p>
|
|
<p id="t9" class="stm run hide_run"><span class="key">from</span> <span class="nam">apscheduler</span><span class="op">.</span><span class="nam">scheduler</span> <span class="key">import</span> <span class="nam">Scheduler</span><span class="strut"> </span></p>
|
|
<p id="t10" class="stm run hide_run"><span class="key">from</span> <span class="op">.</span><span class="nam">models</span> <span class="key">import</span> <span class="nam">Company</span><span class="strut"> </span></p>
|
|
<p id="t11" class="pln"><span class="strut"> </span></p>
|
|
<p id="t12" class="stm run hide_run"><span class="key">import</span> <span class="nam">pandas</span> <span class="key">as</span> <span class="nam">pd</span><span class="strut"> </span></p>
|
|
<p id="t13" class="pln"><span class="strut"> </span></p>
|
|
<p id="t14" class="pln"><span class="com"># LOCAL = False</span><span class="strut"> </span></p>
|
|
<p id="t15" class="pln"><span class="strut"> </span></p>
|
|
<p id="t16" class="stm run hide_run"><span class="key">def</span> <span class="nam">get_hist_predict_data</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t17" class="stm run hide_run"> <span class="nam">recent_data</span><span class="op">,</span><span class="nam">predict_data</span> <span class="op">=</span> <span class="key">None</span><span class="op">,</span><span class="key">None</span><span class="strut"> </span></p>
|
|
<p id="t18" class="pln"> <span class="com"># company = models.Company.objects.get(stock_code=stock_code)</span><span class="strut"> </span></p>
|
|
<p id="t19" class="stm run hide_run"> <span class="nam">company</span> <span class="op">=</span> <span class="nam">get_object_or_404</span><span class="op">(</span><span class="nam">Company</span><span class="op">,</span> <span class="nam">stock_code</span><span class="op">=</span><span class="nam">stock_code</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t20" class="pln"><span class="strut"> </span></p>
|
|
<p id="t21" class="stm run hide_run"> <span class="key">if</span> <span class="nam">company</span><span class="op">.</span><span class="nam">historydata_set</span><span class="op">.</span><span class="nam">count</span><span class="op">(</span><span class="op">)</span> <span class="op"><=</span> <span class="num">0</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t22" class="stm run hide_run"> <span class="nam">history_data</span> <span class="op">=</span> <span class="nam">models</span><span class="op">.</span><span class="nam">HistoryData</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t23" class="stm run hide_run"> <span class="nam">history_data</span><span class="op">.</span><span class="nam">company</span> <span class="op">=</span> <span class="nam">company</span><span class="strut"> </span></p>
|
|
<p id="t24" class="stm run hide_run"> <span class="nam">history_data</span><span class="op">.</span><span class="nam">set_data</span><span class="op">(</span><span class="nam">run</span><span class="op">.</span><span class="nam">get_hist_data</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">=</span><span class="nam">stock_code</span><span class="op">,</span><span class="nam">recent_day</span><span class="op">=</span><span class="num">20</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t25" class="stm run hide_run"> <span class="nam">history_data</span><span class="op">.</span><span class="nam">save</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t26" class="stm run hide_run"> <span class="nam">recent_data</span> <span class="op">=</span> <span class="nam">history_data</span><span class="op">.</span><span class="nam">get_data</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t27" class="pln"> <span class="key">else</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t28" class="stm run hide_run"> <span class="nam">all_data</span> <span class="op">=</span> <span class="nam">company</span><span class="op">.</span><span class="nam">historydata_set</span><span class="op">.</span><span class="nam">all</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t29" class="stm run hide_run"> <span class="key">for</span> <span class="nam">single</span> <span class="key">in</span> <span class="nam">all_data</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t30" class="stm run hide_run"> <span class="nam">now</span> <span class="op">=</span> <span class="nam">dt</span><span class="op">.</span><span class="nam">now</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t31" class="stm run hide_run"> <span class="nam">end_date</span> <span class="op">=</span> <span class="nam">single</span><span class="op">.</span><span class="nam">get_data</span><span class="op">(</span><span class="op">)</span><span class="op">[</span><span class="op">-</span><span class="num">1</span><span class="op">]</span><span class="op">[</span><span class="num">0</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t32" class="stm run hide_run"> <span class="nam">end_date</span> <span class="op">=</span> <span class="nam">dt</span><span class="op">.</span><span class="nam">strptime</span><span class="op">(</span><span class="nam">end_date</span><span class="op">,</span><span class="str">"%Y-%m-%d"</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t33" class="stm run hide_run"> <span class="key">if</span> <span class="nam">now</span><span class="op">.</span><span class="nam">date</span><span class="op">(</span><span class="op">)</span> <span class="op">></span> <span class="nam">end_date</span><span class="op">.</span><span class="nam">date</span><span class="op">(</span><span class="op">)</span><span class="op">:</span> <span class="com"># 更新预测数据</span><span class="strut"> </span></p>
|
|
<p id="t34" class="stm run hide_run"> <span class="nam">single</span><span class="op">.</span><span class="nam">set_data</span><span class="op">(</span><span class="nam">run</span><span class="op">.</span><span class="nam">get_hist_data</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">=</span><span class="nam">stock_code</span><span class="op">,</span><span class="nam">recent_day</span><span class="op">=</span><span class="num">20</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t35" class="stm run hide_run"> <span class="nam">single</span><span class="op">.</span><span class="nam">save</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t36" class="pln"><span class="strut"> </span></p>
|
|
<p id="t37" class="stm run hide_run"> <span class="nam">recent_data</span> <span class="op">=</span> <span class="nam">single</span><span class="op">.</span><span class="nam">get_data</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t38" class="stm run hide_run"> <span class="key">break</span><span class="strut"> </span></p>
|
|
<p id="t39" class="pln"><span class="strut"> </span></p>
|
|
<p id="t40" class="stm run hide_run"> <span class="key">if</span> <span class="nam">company</span><span class="op">.</span><span class="nam">predictdata_set</span><span class="op">.</span><span class="nam">count</span><span class="op">(</span><span class="op">)</span> <span class="op"><=</span> <span class="num">0</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t41" class="stm run hide_run"> <span class="nam">predict_data</span> <span class="op">=</span> <span class="nam">models</span><span class="op">.</span><span class="nam">PredictData</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t42" class="stm run hide_run"> <span class="nam">predict_data</span><span class="op">.</span><span class="nam">company</span> <span class="op">=</span> <span class="nam">company</span><span class="strut"> </span></p>
|
|
<p id="t43" class="stm run hide_run"> <span class="nam">predict_data</span><span class="op">.</span><span class="nam">set_data</span><span class="op">(</span><span class="nam">run</span><span class="op">.</span><span class="nam">prediction</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">,</span><span class="nam">pre_len</span><span class="op">=</span><span class="num">10</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t44" class="stm run hide_run"> <span class="nam">predict_data</span><span class="op">.</span><span class="nam">save</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t45" class="stm run hide_run"> <span class="nam">predict_data</span> <span class="op">=</span> <span class="nam">predict_data</span><span class="op">.</span><span class="nam">get_data</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t46" class="pln"> <span class="key">else</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t47" class="stm run hide_run"> <span class="nam">all_data</span> <span class="op">=</span> <span class="nam">company</span><span class="op">.</span><span class="nam">predictdata_set</span><span class="op">.</span><span class="nam">all</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t48" class="stm run hide_run"> <span class="key">for</span> <span class="nam">single</span> <span class="key">in</span> <span class="nam">all_data</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t49" class="stm run hide_run"> <span class="nam">now</span> <span class="op">=</span> <span class="nam">dt</span><span class="op">.</span><span class="nam">now</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t50" class="stm run hide_run"> <span class="nam">start_date</span> <span class="op">=</span> <span class="nam">dt</span><span class="op">.</span><span class="nam">strptime</span><span class="op">(</span><span class="nam">single</span><span class="op">.</span><span class="nam">start_date</span><span class="op">,</span><span class="str">"%Y-%m-%d"</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t51" class="stm run hide_run"> <span class="key">if</span> <span class="nam">now</span><span class="op">.</span><span class="nam">date</span><span class="op">(</span><span class="op">)</span> <span class="op">></span> <span class="nam">start_date</span><span class="op">.</span><span class="nam">date</span><span class="op">(</span><span class="op">)</span><span class="op">:</span> <span class="com"># 更新预测数据</span><span class="strut"> </span></p>
|
|
<p id="t52" class="stm run hide_run"> <span class="nam">single</span><span class="op">.</span><span class="nam">set_data</span><span class="op">(</span><span class="nam">run</span><span class="op">.</span><span class="nam">prediction</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">,</span> <span class="nam">pre_len</span><span class="op">=</span><span class="num">10</span><span class="op">)</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t53" class="stm run hide_run"> <span class="nam">single</span><span class="op">.</span><span class="nam">save</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t54" class="pln"><span class="strut"> </span></p>
|
|
<p id="t55" class="stm run hide_run"> <span class="nam">predict_data</span> <span class="op">=</span> <span class="nam">single</span><span class="op">.</span><span class="nam">get_data</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t56" class="stm run hide_run"> <span class="key">break</span><span class="strut"> </span></p>
|
|
<p id="t57" class="pln"><span class="strut"> </span></p>
|
|
<p id="t58" class="stm run hide_run"> <span class="key">return</span> <span class="nam">recent_data</span><span class="op">,</span><span class="nam">predict_data</span><span class="strut"> </span></p>
|
|
<p id="t59" class="pln"><span class="strut"> </span></p>
|
|
<p id="t60" class="stm run hide_run"><span class="key">def</span> <span class="nam">get_crawl_save_data</span><span class="op">(</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t61" class="pln"> <span class="str">"""</span><span class="strut"> </span></p>
|
|
<p id="t62" class="pln"><span class="str"> 将10个公司的指标数据爬取并保存到数据库</span><span class="strut"> </span></p>
|
|
<p id="t63" class="pln"><span class="str"> """</span><span class="strut"> </span></p>
|
|
<p id="t64" class="pln"> <span class="com"># 此处应是从网上爬取数据，并保存为csv文件</span><span class="strut"> </span></p>
|
|
<p id="t65" class="stm run hide_run"> <span class="nam">parent_dir</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">dirname</span><span class="op">(</span><span class="nam">__file__</span><span class="op">)</span> <span class="com"># "stock_predict/views.py"</span><span class="strut"> </span></p>
|
|
<p id="t66" class="stm run hide_run"> <span class="nam">file_dir</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">parent_dir</span><span class="op">,</span> <span class="str">"stock_index/"</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t67" class="stm run hide_run"> <span class="key">for</span> <span class="nam">file_name</span> <span class="key">in</span> <span class="nam">os</span><span class="op">.</span><span class="nam">listdir</span><span class="op">(</span><span class="nam">file_dir</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t68" class="stm run hide_run"> <span class="nam">file_path</span> <span class="op">=</span> <span class="nam">os</span><span class="op">.</span><span class="nam">path</span><span class="op">.</span><span class="nam">join</span><span class="op">(</span><span class="nam">file_dir</span><span class="op">,</span> <span class="nam">file_name</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t69" class="stm run hide_run"> <span class="nam">data_frame</span> <span class="op">=</span> <span class="nam">pd</span><span class="op">.</span><span class="nam">read_csv</span><span class="op">(</span><span class="nam">file_path</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t70" class="stm run hide_run"> <span class="nam">stock_code</span> <span class="op">=</span> <span class="nam">file_name</span><span class="op">.</span><span class="nam">split</span><span class="op">(</span><span class="str">'.'</span><span class="op">)</span><span class="op">[</span><span class="num">0</span><span class="op">]</span><span class="strut"> </span></p>
|
|
<p id="t71" class="stm run hide_run"> <span class="nam">company</span> <span class="op">=</span> <span class="nam">get_object_or_404</span><span class="op">(</span><span class="nam">Company</span><span class="op">,</span> <span class="nam">stock_code</span><span class="op">=</span><span class="nam">stock_code</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t72" class="stm run hide_run"> <span class="key">for</span> <span class="nam">index</span><span class="op">,</span><span class="nam">row</span> <span class="key">in</span> <span class="nam">data_frame</span><span class="op">.</span><span class="nam">iterrows</span><span class="op">(</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t73" class="stm run hide_run"> <span class="nam">company</span><span class="op">.</span><span class="nam">stockindex_set</span><span class="op">.</span><span class="nam">create</span><span class="op">(</span><span class="nam">ri_qi</span><span class="op">=</span><span class="nam">row</span><span class="op">[</span><span class="str">'ri_qi'</span><span class="op">]</span><span class="op">,</span><span class="nam">zi_jin</span><span class="op">=</span><span class="nam">row</span><span class="op">[</span><span class="str">'zi_jin'</span><span class="op">]</span><span class="op">,</span><span class="nam">qiang_du</span><span class="op">=</span><span class="nam">row</span><span class="op">[</span><span class="str">'qiang_du'</span><span class="op">]</span><span class="op">,</span><span class="nam">feng_xian</span><span class="op">=</span><span class="nam">row</span><span class="op">[</span><span class="str">'feng_xian'</span><span class="op">]</span><span class="op">,</span><span class="strut"> </span></p>
|
|
<p id="t74" class="pln"> <span class="nam">zhuan_qiang</span><span class="op">=</span><span class="nam">row</span><span class="op">[</span><span class="str">'zhuan_qiang'</span><span class="op">]</span><span class="op">,</span><span class="nam">chang_yu</span><span class="op">=</span><span class="nam">row</span><span class="op">[</span><span class="str">'chang_yu'</span><span class="op">]</span><span class="op">,</span><span class="nam">jin_zi</span><span class="op">=</span><span class="nam">row</span><span class="op">[</span><span class="str">'jin_zi'</span><span class="op">]</span><span class="op">,</span><span class="nam">zong_he</span><span class="op">=</span><span class="nam">row</span><span class="op">[</span><span class="str">'zong_he'</span><span class="op">]</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t75" class="pln"><span class="strut"> </span></p>
|
|
<p id="t76" class="stm run hide_run"><span class="key">def</span> <span class="nam">get_stock_index</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t77" class="pln"> <span class="str">"""</span><span class="strut"> </span></p>
|
|
<p id="t78" class="pln"><span class="str"> 获取股票的各项指标数据</span><span class="strut"> </span></p>
|
|
<p id="t79" class="pln"><span class="str"> """</span><span class="strut"> </span></p>
|
|
<p id="t80" class="stm run hide_run"> <span class="nam">company</span> <span class="op">=</span> <span class="nam">get_object_or_404</span><span class="op">(</span><span class="nam">Company</span><span class="op">,</span> <span class="nam">stock_code</span><span class="op">=</span><span class="nam">stock_code</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t81" class="stm run hide_run"> <span class="key">if</span> <span class="nam">company</span><span class="op">.</span><span class="nam">stockindex_set</span><span class="op">.</span><span class="nam">count</span><span class="op">(</span><span class="op">)</span> <span class="op"><=</span> <span class="num">0</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t82" class="pln"> <span class="com"># 将爬取的数据存入数据库</span><span class="strut"> </span></p>
|
|
<p id="t83" class="stm mis"> <span class="nam">get_crawl_save_data</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t84" class="pln"> <span class="com"># 从数据库获取近三天的数据</span><span class="strut"> </span></p>
|
|
<p id="t85" class="stm run hide_run"> <span class="nam">indexs</span> <span class="op">=</span> <span class="nam">company</span><span class="op">.</span><span class="nam">stockindex_set</span><span class="op">.</span><span class="nam">all</span><span class="op">(</span><span class="op">)</span><span class="op">.</span><span class="nam">order_by</span><span class="op">(</span><span class="str">'-ri_qi'</span><span class="op">)</span><span class="op">[</span><span class="op">:</span><span class="num">3</span><span class="op">]</span><span class="op">.</span><span class="nam">values</span><span class="op">(</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t86" class="stm run hide_run"> <span class="key">return</span> <span class="nam">list</span><span class="op">(</span><span class="nam">indexs</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t87" class="pln"><span class="strut"> </span></p>
|
|
<p id="t88" class="pln"><span class="strut"> </span></p>
|
|
<p id="t89" class="stm run hide_run"><span class="key">def</span> <span class="nam">home</span><span class="op">(</span><span class="nam">request</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t90" class="stm run hide_run"> <span class="nam">recent_data</span><span class="op">,</span><span class="nam">predict_data</span> <span class="op">=</span> <span class="nam">get_hist_predict_data</span><span class="op">(</span><span class="str">"600718"</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t91" class="stm run hide_run"> <span class="nam">data</span> <span class="op">=</span> <span class="op">{</span><span class="str">"recent_data"</span><span class="op">:</span><span class="nam">recent_data</span><span class="op">,</span><span class="str">"stock_code"</span><span class="op">:</span><span class="str">"600718"</span><span class="op">,</span><span class="str">"predict_data"</span><span class="op">:</span><span class="nam">predict_data</span><span class="op">}</span><span class="strut"> </span></p>
|
|
<p id="t92" class="stm run hide_run"> <span class="nam">data</span><span class="op">[</span><span class="str">'indexs'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">get_stock_index</span><span class="op">(</span><span class="str">"600718"</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t93" class="stm run hide_run"> <span class="key">return</span> <span class="nam">render</span><span class="op">(</span><span class="nam">request</span><span class="op">,</span><span class="str">"stock_predict/home.html"</span><span class="op">,</span><span class="op">{</span><span class="str">"data"</span><span class="op">:</span><span class="nam">json</span><span class="op">.</span><span class="nam">dumps</span><span class="op">(</span><span class="nam">data</span><span class="op">)</span><span class="op">}</span><span class="op">)</span> <span class="com"># json.dumps(list)</span><span class="strut"> </span></p>
|
|
<p id="t94" class="pln"><span class="strut"> </span></p>
|
|
<p id="t95" class="stm run hide_run"><span class="key">def</span> <span class="nam">predict_stock_action</span><span class="op">(</span><span class="nam">request</span><span class="op">)</span><span class="op">:</span><span class="strut"> </span></p>
|
|
<p id="t96" class="stm run hide_run"> <span class="nam">stock_code</span> <span class="op">=</span> <span class="nam">request</span><span class="op">.</span><span class="nam">POST</span><span class="op">.</span><span class="nam">get</span><span class="op">(</span><span class="str">'stock_code'</span><span class="op">,</span><span class="key">None</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t97" class="pln"> <span class="com"># print("stock_code:\n",stock_code)</span><span class="strut"> </span></p>
|
|
<p id="t98" class="stm run hide_run"> <span class="nam">recent_data</span><span class="op">,</span> <span class="nam">predict_data</span> <span class="op">=</span> <span class="nam">get_hist_predict_data</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t99" class="stm run hide_run"> <span class="nam">data</span> <span class="op">=</span> <span class="op">{</span><span class="str">"recent_data"</span><span class="op">:</span> <span class="nam">recent_data</span><span class="op">,</span> <span class="str">"stock_code"</span><span class="op">:</span> <span class="nam">stock_code</span><span class="op">,</span> <span class="str">"predict_data"</span><span class="op">:</span> <span class="nam">predict_data</span><span class="op">}</span><span class="strut"> </span></p>
|
|
<p id="t100" class="stm run hide_run"> <span class="nam">data</span><span class="op">[</span><span class="str">'indexs'</span><span class="op">]</span> <span class="op">=</span> <span class="nam">get_stock_index</span><span class="op">(</span><span class="nam">stock_code</span><span class="op">)</span><span class="strut"> </span></p>
|
|
<p id="t101" class="stm run hide_run"> <span class="key">return</span> <span class="nam">render</span><span class="op">(</span><span class="nam">request</span><span class="op">,</span> <span class="str">"stock_predict/home.html"</span><span class="op">,</span> <span class="op">{</span><span class="str">"data"</span><span class="op">:</span> <span class="nam">json</span><span class="op">.</span><span class="nam">dumps</span><span class="op">(</span><span class="nam">data</span><span class="op">)</span><span class="op">}</span><span class="op">)</span> <span class="com"># json.dumps(list)</span><span class="strut"> </span></p>
|
|
<p id="t102" class="pln"><span class="strut"> </span></p>
|
|
<p id="t103" class="stm run hide_run"><span class="str">'''</span><span class="strut"> </span></p>
|
|
<p id="t104" class="pln"><span class="str">sched = Scheduler()</span><span class="strut"> </span></p>
|
|
<p id="t105" class="pln"><span class="str"># 定时任务</span><span class="strut"> </span></p>
|
|
<p id="t106" class="pln"><span class="str"># @sched.interval_schedule(seconds=2) # 每2s执行一次</span><span class="strut"> </span></p>
|
|
<p id="t107" class="pln"><span class="str">@sched.cron_schedule(hour=0,minute=0) # 每日凌晨调度一次</span><span class="strut"> </span></p>
|
|
<p id="t108" class="pln"><span class="str">def train_models():</span><span class="strut"> </span></p>
|
|
<p id="t109" class="pln"><span class="str"> run.train_all_stock()</span><span class="strut"> </span></p>
|
|
<p id="t110" class="pln"><span class="strut"> </span></p>
|
|
<p id="t111" class="pln"><span class="str">sched.start()</span><span class="strut"> </span></p>
|
|
<p id="t112" class="pln"><span class="str">'''</span><span class="strut"> </span></p>
|
|
|
|
</td>
|
|
</tr>
|
|
</table>
|
|
</div>
|
|
|
|
<div id="footer">
|
|
<div class="content">
|
|
<p>
|
|
<a class="nav" href="index.html">« index</a> <a class="nav" href="https://coverage.readthedocs.io">coverage.py v4.5.2</a>,
|
|
created at 2018-12-22 16:12
|
|
</p>
|
|
</div>
|
|
</div>
|
|
|
|
</body>
|
|
</html>
|