Commit 11aef5d

committed

defined DF at beginning of script

1 parent 7ead2a5 commit 11aef5dCopy full SHA for 11aef5d

File tree

-2

lines changed

-2

lines changed

Lines changed: 3 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -23,6 +23,9 @@`
`23`	`23`	`parser.add_option("-c", "--csv", action="store_true", dest="csv", help="Saves extracted contents to a CSV file.")`
`24`	`24`	`parser.add_option("-m", "--mongo", action="store_true", dest="mongo", help="Saves extracted contents to a MongoDB.")`
`25`	`25`
	`26`	`+# Defined DataFrame to avoid check errors`
	`27`	`+df = pd.DataFrame()`
	`28`	`+`
`26`	`29`	`# MongoDB Collection (Table) Template`
`27`	`30`	`class Trending(mongoengine.Document):`
`28`	`31`	`section = mongoengine.StringField(required=True)`
`@@ -132,8 +135,6 @@ def save_to_csv():`
`132`	`135`
`133`	`136`	`if mongo:`
`134`	`137`	`mongoengine.connect("Youtube")`
`135`		`- if csv:`
`136`		`- df = pd.DataFrame()`
`137`	`138`
`138`	`139`	`driver = load_driver() # load driver`
`139`	`140`	`page_scrap(driver) # start scrapping`

Comments

(0)