add notebook wrapper for normal file jobs

bundle/python/notebook_task_wrappers.go

@@ -89,11 +89,14 @@ func (n *notebookTrampoline) GetTemplate(b *bundle.Bundle, task *jobs.Task) (str
 
 	lines := strings.Split(s, "\n")
 	if strings.HasPrefix(lines[0], "# Databricks notebook source") {
-		return getDbnbTemplate(strings.Join(lines[1:], "\n"))
+		return getDbnbTemplate(strings.Join(lines, "\n"))
 	}
 
-	//TODO return getPyTemplate(s), nil
-	return s, nil
+	return getPyTemplate(s), nil
+}
+
+func getPyTemplate(s string) string {
+	return pyTrampolineData
 }
 
 func getDbnbTemplate(s string) (string, error) {

bundle/python/trampoline_data/python.py

@@ -3,7 +3,7 @@ import runpy
 import sys
 import os
 
-python_file = {{.PythonFile}}
+python_file = {{.SourceFile}}
 project_root = {{.ProjectRoot}}
 #remove databricks args from argv
 sys.argv = sys.argv[1:]