Skip to content

Commit 7ecefac

Browse files
committed
Fixed bug on create_sketch, implemented copy_sketch, added .gitignore.
Signed-off-by: ubi de feo <[email protected]>
1 parent 144813e commit 7ecefac

File tree

2 files changed

+15
-4
lines changed

2 files changed

+15
-4
lines changed

.gitignore

+2
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
.DS_Store
2+
*.mpy

arduino/arduino.py

+13-4
Original file line numberDiff line numberDiff line change
@@ -82,21 +82,26 @@ def delay(_ms):
8282

8383

8484
# HELPERS
85-
def create_sketch(sketch_name = None, path = '.', overwrite = False):
85+
86+
def get_template():
87+
return '/'.join(__file__.split('/')[:-1]) + '/template.py'
88+
89+
90+
def create_sketch(sketch_name = None, destination_path = '.', overwrite = False, source = None):
8691

8792
if sketch_name is None:
8893
sketch_name = 'main'
89-
new_sketch_path = f'{path}/{sketch_name}.py'
94+
new_sketch_path = f'{destination_path}/{sketch_name}.py'
9095
try:
9196
open(new_sketch_path, 'r')
9297
if not overwrite:
9398
sketch_name = f'{sketch_name}_{ticks_us()}'
9499
except OSError:
95100
pass
96101

97-
template_path = '/'.join(__file__.split('/')[:-1]) + '/template.py'
102+
template_path = get_template() if source is None else source
98103
template_sketch = open(template_path, 'r')
99-
new_sketch_path = f'{path}/{sketch_name}.py'
104+
new_sketch_path = f'{destination_path}/{sketch_name}.py'
100105

101106
with open(new_sketch_path, 'w') as f:
102107
sketch_line = None
@@ -107,6 +112,10 @@ def create_sketch(sketch_name = None, path = '.', overwrite = False):
107112
template_sketch.close()
108113
return new_sketch_path
109114

115+
def copy_sketch(source_path = '', destination_path = '.', name = None, overwrite = False):
116+
name = name or 'main'
117+
return create_sketch(sketch_name = name, destination_path = destination_path, overwrite = overwrite, source = source_path)
118+
110119
# the following methods are just for testing
111120
# will produce output when this module is run as __main__
112121
def preload():

0 commit comments

Comments
 (0)